Update app.py

app.py

@@ -14,6 +14,8 @@ classifier = pipeline("zero-shot-classification", model="MoritzLaurer/mDeBERTa-v
 #     titles_astiko = pickle.load(file)
 labels1 = ["κληρονομικό", "εμπορικό", "διαζύγιο"]
 labels2 = ["αποδοχή κληρονομιάς", "κληρονόμοι", "ιδιόγραφη διαθήκη"]
+labels3 = ["ερώτηση", "απαιτούμενα έγγραφα"]
+
 
 # titles_astiko = ["γάμος", "αλλοδαπός", "φορολογία", "κληρονομικά", "στέγη", "οικογενειακό", "εμπορικό","κλοπή","απάτη"]
 
@@ -21,6 +23,7 @@ labels2 = ["αποδοχή κληρονομιάς", "κληρονόμοι", "ι
 def classify(text):
     output = classifier(text, labels1, multi_label=False)
     output2 = classifier(text, labels2, multi_label=False)
+    output3 = classifier(text, labels3, multi_label=False)
     
     
     # for i in range(len(scores)):
@@ -30,11 +33,12 @@ def classify(text):
     #         break
     # # output = output[0:10]
     # return return_labels
-    return output, output2
+    return output, output2, output3
 
 
 text = st.text_input('Enter some text:')  # Input field for new text
 if text:
-    output1, output2 = classify(text)
+    output1, output2, output3 = classify(text)
     st.text(output1)
-    st.text(output2)
+    st.text(output2)
+    st.text(output3)