Update app.py

app.py

@@ -12,7 +12,7 @@ classifier = pipeline("zero-shot-classification", model="MoritzLaurer/mDeBERTa-v
 
 # with open('chapter_titles.pkl', 'rb') as file:
 #     titles_astiko = pickle.load(file)
-# labels1 = ["κληρονομικό", "ακίνητα", "διαζύγιο"]
+labels1 = ["κληρονομικό", "ακίνητα", "διαζύγιο"]
 # labels2 = ["αποδοχή κληρονομιάς", "αποποίηση", "διαθήκη"]
 # labels3 = ["μίσθωση", "κυριότητα", "έξωση", "απλήρωτα νοίκια"]
 
@@ -32,23 +32,27 @@ text = st.text_input('Enter some text:')  # Input field for new text
 
 if text:
 
-    labels = list(dictionary)
+    # labels = list(dictionary)
     
-    output = classify(text,labels)
+    # output = classify(text,labels)
 
-    output = output["labels"][0]
+    # output = output["labels"][0]
 
-    labels = list(dictionary[output])
+    # labels = list(dictionary[output])
 
-    output2 = classify(text,labels)
+    # output2 = classify(text,labels)
 
-    output2 = output2["labels"][0]
+    # output2 = output2["labels"][0]
 
 
-    answer = dictionary[output][output2]
+    # answer = dictionary[output][output2]
 
-    st.text(output)
-    st.text(output2)
+    # st.text(output)
+    # st.text(output2)
+
+    # st.text(answer)
 
-    st.text(answer)
+    
+    output = classify(text,labels1)
+    st.text(output)