first commit

app.py

@@ -56,22 +56,21 @@ def prep_examples():
     example_labels11 = "bhalo;;kharap"
 
     examples = [
-        [example_text1, example_labels1],
-        [example_text2, example_labels2],
-        [example_text3, example_labels3],
-        [example_text4, example_labels4],
-        [example_text5, example_labels5],
-        [example_text6, example_labels6],
-        [example_text7, example_labels7],
-        [example_text8, example_labels8],
-        [example_text9, example_labels9],
-        [example_text10, example_labels10],
-        [example_text11, example_labels11]]
+        [example_text1, example_labels1, "{}"],
+        [example_text2, example_labels2, "{}"],
+        [example_text3, example_labels3, "{}"],
+        [example_text4, example_labels4, "{}"],
+        [example_text5, example_labels5, "{}"],
+        [example_text6, example_labels6, "{}"],
+        [example_text7, example_labels7, "{}"],
+        [example_text8, example_labels8, "{}"],
+        [example_text9, example_labels9, "{}"],
+        [example_text10, example_labels10, "{}"],
+        [example_text11, example_labels11, "{}"]]
 
     return examples
 
 def detect_lang(text):
-    DetectorFactory.seed = 0
     seq_lang = 'en'
     
     text = text.replace('\n', ' ')
@@ -86,15 +85,17 @@ def detect_lang(text):
 
     return seq_lang
 
-def sequence_to_classify(text, labels):
+def sequence_to_classify(text, labels, hypothesis_template):
 	lang = detect_lang(text)
+	
+	print("hypothesis_template", hypothesis_template)
 	if lang == 'en':
 		model = model_en
 		hypothesis_template = "{}"
 	else:
 		model = model_multi
 		hypothesis_template = "{}"
-	
+
 	labels = [hypothesis_template.format(label) for label in labels.split(";;")]
 	
 	if str(type(model)) == "<class 'clip.model.CLIP'>":
@@ -124,6 +125,7 @@ def sequence_to_classify(text, labels):
 		pred["scores"], pred["labels"] = zip(*sorted_sl)
 		preds.append(pred)
 	
+	print(preds)
 	if len(preds) == 1:
 	  preds = preds[0]
 
@@ -148,7 +150,10 @@ iface = gr.Interface(
         placeholder="Text here..."),
         gr.inputs.Textbox(lines=2,
         label="Please enter the candidate labels (separated by 2 consecutive semicolons)...",
-        placeholder="Labels here separated by ;;")],
+        placeholder="Labels here separated by ;;"),
+        gr.inputs.Textbox(lines=2,
+        label="Please enter the text for hypothesis template(optional)...",
+        placeholder="Text here...")],
     outputs=gr.outputs.Label(num_top_classes=5),
     #interpretation="default",
     examples=prep_examples())