Update app.py

app.py

@@ -12,34 +12,35 @@ classifier = pipeline("zero-shot-classification", model="MoritzLaurer/mDeBERTa-v
 
 # with open('chapter_titles.pkl', 'rb') as file:
 #     titles_astiko = pickle.load(file)
-labels1 = ["κληρονομικό", "ακίνητα", "διαζύγιο"]
-labels2 = ["αποδοχή κληρονομιάς", "αποποίηση", "διαθήκη"]
-labels3 = ["μίσθωση", "κυριότητα", "έξωση", "απλήρωτα νοίκια"]
-
+# labels1 = ["κληρονομικό", "ακίνητα", "διαζύγιο"]
+# labels2 = ["αποδοχή κληρονομιάς", "αποποίηση", "διαθήκη"]
+# labels3 = ["μίσθωση", "κυριότητα", "έξωση", "απλήρωτα νοίκια"]
 
 
 # titles_astiko = ["γάμος", "αλλοδαπός", "φορολογία", "κληρονομικά", "στέγη", "οικογενειακό", "εμπορικό","κλοπή","απάτη"]
+# Load dictionary from the file using pickle
+with open('my_dict.pickle', 'rb') as file:
+    dictionary = pickle.load(file)
 
-
-def classify(text):
+def classify(text,labels):
     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)):
-    #     if scores[i] > 0.99:
-    #         return_labels.append(labels[i])
-    #     else:
-    #         break
-    # # output = output[0:10]
-    # return return_labels
-    return output, output2, output3
+    return output
 
 
 text = st.text_input('Enter some text:')  # Input field for new text
+
 if text:
-    output1, output2, output3 = classify(text)
-    st.text(output1)
-    st.text(output3)
-    st.text(output2)
+
+    labels = list(dictionary)
+    
+    output = classify(text,labels)
+
+    labels = list(dictionary[output])
+
+    output2 = classify(text,labels)
+
+    answer = dictionary[output][output2]
+
+    st.text(answer)
+