this

app.py

@@ -10,8 +10,9 @@ from transformers import pipeline, Conversation
 
 
 
-classifyr = pipeline("zero-shot-classification")
-convo = pipeline("conversational")
+# classifyr = pipeline("zero-shot-classification")
+# convo = pipeline("conversational")
+classifi = pipeline(model="facebook/bart-large-mnli")
 
 uri = os.environ["MONGO_CONNECTION_STRING"]
 client = MongoClient(uri, tlsCertificateKeyFile="database/cert.pem") 
@@ -33,7 +34,7 @@ def view_rem():
 def Chatbot():
     st.title("Chatbot")
     if query :=st.chat_input("Enter your message"):
-        ans = classifyr(query,candidate_labels=["reminders", "general conversation"])
+        ans = classifi(query,candidate_labels=["reminders", "general conversation"])
         if ans["labels"][0] == "reminders":
             values = getValues(query.lower())
             with st.chat_message("assistant"):
@@ -42,7 +43,7 @@ def Chatbot():
                 
                 
         elif ans["labels"][0] == "general conversation":
-            umsg = bard.get_answer(message)["content"]
+            umsg = bard.get_answer(query)["content"]
             with st.chat_message("assistant"):
                 st.write(umsg)