Trying zero-shot classification.

app.py

@@ -1,14 +1,15 @@
 import gradio
 import json
 from transformers import pipeline
+from transformers import AutoTokenizer
 
 def zero_shot_classification(data_string):
     print(data_string)
     data = json.loads(data_string)
     print(data)
-    classifier = pipeline('zero-shot-classification', model='Xenova/mobilebert-uncased-mnli')
+    classifier = pipeline(task='zero-shot-classification', tokenizer=AutoTokenizer.from_pretrained('Xenova/mobilebert-uncased-mnli'), model='Xenova/mobilebert-uncased-mnli')
 
-    results = classifier(data.sequence, candidate_labels=data.candidate_labels, hypothesis_template=data.hypothesis_template, multi_label=data.multi_label)
+    results = classifier(data.sequence, candidate_labels=data.candidate_labels) #, hypothesis_template=data.hypothesis_template, multi_label=data.multi_label)
     return {results}
 
 gradio_interface = gradio.Interface(