Update app.py

app.py

@@ -64,6 +64,9 @@ response_model_name = "microsoft/DialoGPT-medium"
 response_tokenizer = AutoTokenizer.from_pretrained(response_model_name)
 response_model = AutoModelForCausalLM.from_pretrained(response_model_name)
 
+# Set the pad token
+response_tokenizer.pad_token = response_tokenizer.eos_token
+
 # Enhanced Emotional States
 emotions = {
     'joy': {'percentage': 10, 'motivation': 'positive', 'intensity': 0},
@@ -148,7 +151,6 @@ def evolve_emotions():
     total = sum(e['percentage'] for e in emotions.values())
     for e in emotions:
         emotions[e]['percentage'] = (emotions[e]['percentage'] / total) * 100
-
 def update_emotion_history(emotion, percentage, intensity, context):
     entry = {
         'emotion': emotion,
@@ -198,7 +200,8 @@ def generate_response(input_text, ai_emotion):
             do_sample=True,
             top_k=50,
             top_p=0.95,
-            temperature=temperature
+            temperature=temperature,
+            pad_token_id=response_tokenizer.eos_token_id
         )
     response = response_tokenizer.decode(response_ids[0], skip_special_tokens=True)
     
@@ -326,4 +329,4 @@ iface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    iface.launch(share=True)
+    iface.launch(share=True)