Update app/response_generator.py

app/response_generator.py

@@ -29,6 +29,12 @@ def generate_support_message(text: str, emotion: str) -> str:
         outputs = model.generate(
             **inputs, max_new_tokens=80, do_sample=True, temperature=0.7
         )
+    
+    decoded = tokenizer.decode(outputs[0], skip_special_tokens=True)
+
+    if "Assistant:" in decoded:
+        response = decoded.split("Assistant:")[-1].strip()
+    else:
+        response = decoded.strip()
 
-    response = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return response