Update app.py

app.py

@@ -23,6 +23,9 @@ def list2text(label_list):
 label_list_1 = ["dünya", "ekonomi", "kültür", "sağlık", "siyaset", "spor", "teknoloji"]
 label_list_2 = ["positive", "negative", "neutral"]
 
+hypothesis_1 = "Bu yazı {} konusundadır"
+hypothesis_2 = "This text is in {} subject"
+
 st.title("Multilingual Zero-Shot Text Classification with mT5")
 
 model_name = "alan-turing-institute/mt5-large-finetuned-mnli-xtreme-xnli"
@@ -38,9 +41,12 @@ st.text_area("Text #1", text_1, height=128)
 st.text_area("Text #2", text_2, height=128)
 st.write(f"Label List #1: {list2text(label_list_1)}")
 st.write(f"Label List #2: {list2text(label_list_2)}")
+st.write(f"Hypothesis #1: {hyphothesis_1}")
+st.write(f"Hypothesis #2: {hyphothesis_2}")
 
 text = st.radio("Select Text", ("Text #1", "Text #2", "New Text"))
 labels = st.radio("Select Label List", ("Label List #1", "Label List #2", "New Label List"))
+hyphothesis = st.radio("Select Hyphothesis", ("Hyphothesis #1", "Hyphothesis #2", "New Hyphothesis"))
 
 if text == "Text #1": sequence_to_classify = text_1
 elif text == "Text #2": sequence_to_classify = text_2
@@ -52,7 +58,10 @@ elif labels == "Label List #2": candidate_labels = label_list_2
 elif labels == "New Label List":
     candidate_labels = st.text_area("New Label List (Pls Input as comma-separated)", value="", height=16).split(",")
 
-hypothesis_template = "Bu yazı {} konusundadır."
+if hyphothesis == "Hyphothesis #1": hyphothesis_template = hyphothesis_1
+elif hyphothesis == "Hyphothesis #2": hyphothesis_template = hyphothesis_2
+elif labels == "New Hyphothesis":
+    hyphothesis_template = st.text_area("Hypothesis Template for NLI (Pls Input as comma-separated)", value="", height=16)
         
 Run_Button = st.button("Run", key=None)
 if Run_Button == True: