Update app.py

app.py

@@ -3,3 +3,20 @@ import gradio as gr
 
 classifier = pipeline("zero-shot-classification", model="DeepPavlov/xlm-roberta-large-en-ru-mnli")
 
+def wrap_classifier(text, labels, template):
+    labels = labels.split(",")
+    outputs = classifier(text, labels, hypothesis_template=template)
+    return outputs["labels"][0]
+
+gr.Interface(
+    fn=wrap_classifier,
+    title="Zero-shot Classification",
+    inputs=[
+        gr.inputs.Textbox(lines=5, label="Text to classify"),
+        gr.inputs.Textbox(lines=1, label="Candidate labels separated with commas"),
+        gr.inputs.Textbox(lines=1, label="Template", default="This sentence is about {}.", placeholder="This sentence is about {}.")
+    ],
+    outputs=[
+        gr.outputs.Textbox(label="Label")
+    ],
+).launch()