Spaces:
Runtime error
Runtime error
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()
|