Update app.py

app.py

@@ -174,7 +174,7 @@ def predict_emotion(context):
 def generate_text(prompt, chat_history, emotion=None, max_length=100):
     finetuned_lm_tokenizer, finetuned_lm_model = get_finetuned_lm_model()
     
-     full_prompt = ""
+    full_prompt = ""
     for turn in chat_history[-5:]:  # Consider last 5 turns for context
         full_prompt += f"{finetuned_lm_tokenizer.eos_token}{turn[0]}{finetuned_lm_tokenizer.eos_token}{turn[1]}"
     full_prompt += f"{finetuned_lm_tokenizer.eos_token}{prompt}"
@@ -205,8 +205,7 @@ def generate_text(prompt, chat_history, emotion=None, max_length=100):
         do_sample=True,
         temperature=temperature,
         top_k=top_k,
-        top_p=top_p,
-        attention_mask=attention_mask
+        top_p=top_p
     )
 
     generated_text = finetuned_lm_tokenizer.decode(output[0], skip_special_tokens=True)
@@ -255,7 +254,7 @@ def respond_to_user(user_input, chat_history):
     dominant_emotion = get_dominant_emotion()
     
     # Generate a response considering the dominant emotion
-    response = generate_text(user_input, dominant_emotion)
+    response = generate_text(user_input, chat_history, dominant_emotion)
     
     # Update emotion history
     update_emotion_history(predicted_emotion, emotions[predicted_emotion]['intensity'])