File size: 591 Bytes
6ef3a78
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
import gradio as gr
from transformers import pipeline

model = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")


def predict(field_name, property_list):
    properties = property_list.split(",")
    results = model(field_name, properties)
    top_results = []
    for i, result in enumerate(results["labels"]):
        if i < 10:
            top_results.append(f"{result}: {results['scores'][i]}")
    return results["labels"][0], "\n".join(top_results)


demo = gr.Interface(
    fn=predict, inputs=["text", "textarea"], outputs=["text", "textarea"]
)

demo.launch()