kumaranke commited on
Commit
ae1fc75
1 Parent(s): 460172f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -0
app.py CHANGED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import pipeline
2
+ import gradio as grad
3
+
4
+ zero_shot_classifier = pipeline("zero-shot-classification")
5
+
6
+ def classify(text, labels):
7
+ clssifier_labels = labels.splits(",")
8
+ #["software","politics","love","movies","emergency","advertisement","sports"]
9
+ response = zero_shot_classifier(text, classifier_labels)
10
+ return response
11
+
12
+ txt = grad.Textbox(lines=1, label="English", placeholder="text to be classified")
13
+ labels = grad.Textbox(lines=1, label="Labels", placeholder = "comma separated labels")
14
+ out = grad.Textbox(lines=1, label = "Classification")
15
+
16
+ grad.Interface(classify, inputs=[txt, labels], outputs = out).launch()