kmontag commited on
Commit
6ef3a78
1 Parent(s): 2120536

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +21 -0
app.py ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import pipeline
3
+
4
+ model = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
5
+
6
+
7
+ def predict(field_name, property_list):
8
+ properties = property_list.split(",")
9
+ results = model(field_name, properties)
10
+ top_results = []
11
+ for i, result in enumerate(results["labels"]):
12
+ if i < 10:
13
+ top_results.append(f"{result}: {results['scores'][i]}")
14
+ return results["labels"][0], "\n".join(top_results)
15
+
16
+
17
+ demo = gr.Interface(
18
+ fn=predict, inputs=["text", "textarea"], outputs=["text", "textarea"]
19
+ )
20
+
21
+ demo.launch()