Update app.py

app.py

@@ -7,8 +7,10 @@ import tempfile
 from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
 
 # Load chatbot model
-tokenizer = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-3B")
-chatbot = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-3B")
+from transformers import pipeline
+
+chatbot = pipeline("conversational", model="facebook/blenderbot_small-90M")
+
 
 # Load emotion model
 model = load_model("emotion_model.h5")
@@ -21,10 +23,12 @@ def detect_emotion(img):
     return emotion_labels[np.argmax(prediction)]
 
 def get_chatbot_reply(user_message):
-    inputs = tokenizer([user_message], return_tensors="pt")
-    reply_ids = chatbot.generate(**inputs)
-    reply = tokenizer.decode(reply_ids[0], skip_special_tokens=True)
-    return reply
+   from transformers import Conversation
+
+def get_chatbot_reply(user_message):
+    conversation = Conversation(user_message)
+    result = chatbot(conversation)
+    return result.generated_responses[-1]
 
 def smart_mirror(image, user_input):
     emotion = detect_emotion(image)