micole66 commited on
Commit
8e735f3
1 Parent(s): 31f495e

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -0
app.py ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import pipeline
3
+
4
+ classifier = pipeline("zero-shot-classification", model="MoritzLaurer/DeBERTa-v3-large-mnli-fever-anli-ling-wanli")
5
+
6
+ def zeroShotClassification(text_input, candidate_labels):
7
+ labels = [label.strip(' ') for label in candidate_labels.split(',')]
8
+ output = {}
9
+ prediction = classifier(text_input, labels)
10
+ for i in range(len(prediction['labels'])):
11
+ output[prediction['labels'][i]] = prediction['scores'][i]
12
+ return output
13
+
14
+ examples = [["One day I will see the world", "travel, live, die, future"]]
15
+
16
+ demo = gr.Interface(fn=zeroShotClassification, inputs=["text", "text"], outputs="label", title="Text Classification", examples=examples)
17
+ demo.launch()