File size: 542 Bytes
4f77bf2
6c382fe
9b18155
9b135ca
 
4f77bf2
9b135ca
6c382fe
d9321e0
 
 
 
4f77bf2
d9321e0
4f77bf2
d9321e0
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
from transformers import pipeline
import gradio as gr
import os
token = os.environ["token"]

classifier = pipeline("zero-shot-classification",
                      model="joeddav/xlm-roberta-large-xnli",use_auth_token=token)

def classify(text, intent_labels):
    res = classifier(text, intent_labels)
    results = {k: v for k, v in zip(res["labels"], res["scores"])}
    return results

# input is two text boxes, one for the text and one for the labels

gr.Interface(fn=classify, inputs=["textbox", "textbox"], outputs="label").launch()