kmontag commited on
Commit
2120536
1 Parent(s): f180eef

Delete demo.py

Browse files
Files changed (1) hide show
  1. demo.py +0 -21
demo.py DELETED
@@ -1,21 +0,0 @@
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()