Update app.py

app.py

@@ -48,25 +48,19 @@ labels = st.radio("Select Label List", ("Label List #1", "Label List #2", "New L
 if text == "Text #1": selected_text = text_1
 elif text == "Text #2": selected_text = text_2
 elif text == "New Text":
-    selected_text = st.text_area("New Text", value="", height=128)
+    sequence_to_classify = st.text_area("New Text", value="", height=128)
 
 if labels == "Label List #1": selected_labels = label_list_1
 elif labels == "Label List #2": selected_labels = label_list_2
 elif labels == "New Label List":
-    selected_labels = st.text_area("New Label List (Pls Input as comma-separated)", value="", height=16).split(",")
+    candidate_labels = st.text_area("New Label List (Pls Input as comma-separated)", value="", height=16).split(",")
+
+hypothesis_template = "Bu yazı {} konusundadır."
 
-@st.cache(allow_output_mutation=True)
-def setModel(model_name):
-    tokenizer = MT5Tokenizer.from_pretrained(model_name)
-    model = MT5ForConditionalGeneration.from_pretrained(model_name)
-    model.eval()
-    return pipeline("zero-shot-classification", model=model, tokenizer=tokenizer)
         
 Run_Button = st.button("Run", key=None)
 if Run_Button == True:
-    
-    zstc_pipeline = setModel(model_checkpoint)
-    output = zstc_pipeline(sequences=selected_text, candidate_labels=selected_labels)
+    output = runModel(model_name, sequence_to_classify, candidate_labels, hypothesis_template)
     output_labels = output["labels"]
     output_scores = output["scores"]