Update app.py

app.py

@@ -23,7 +23,7 @@ with open('my_dict.pickle', 'rb') as file:
     dictionary = pickle.load(file)
 
 def classify(text,labels):
-    output = classifier(text, labels1, multi_label=False)
+    output = classifier(text, labels, multi_label=False)
     
     return output
 
@@ -36,10 +36,15 @@ if text:
     
     output = classify(text,labels)
 
+    output = output["labels"][0]
+
     labels = list(dictionary[output])
 
     output2 = classify(text,labels)
 
+    output2 = output2["labels"][0]
+
+
     answer = dictionary[output][output2]
 
     st.text(answer)