Update app.py

app.py

@@ -5,10 +5,9 @@ client = MongoClient("mongodb+srv://Savir:savir2010@m0-savir.8iqdy.mongodb.net/?
 
 db = client['sentiment-db']
 collection = db['sentiment-collection']
-results = collection.find({"name":'savir'})
-for result in results:
-    st.markdown(f"{result}")
-exit()
+# results = collection.find({"name":'savir'})
+# for result in results:
+#     st.markdown(f"{result}")
 from peft import PeftModel, PeftConfig
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 import torch
@@ -37,4 +36,6 @@ predictions = torch.max(logits,1).indices
 if st.button("Calculate Sentiment"):
     with st.chat_message("assistant"):
         response = "The Following Text " + f'"{review}" is ' + id2label[predictions.tolist()[0]]
+        response_dict = {review:id2label[predictions.tolist()[0]]}
+        response_db = collection.insert_one(response_dict)
         st.markdown(response)