Update app.py

app.py

@@ -76,9 +76,10 @@ def generate_response(user_input, model_id):
         response = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
         # Clean up response
-        cleaned_response = re.sub(r"(User:|Assistant:)", "", response).strip()
-        return cleaned_response.split("\n")[0]
-
+        #cleaned_response = re.sub(r"(User:|Assistant:)", "", response).strip()
+        #return cleaned_response.split("\n")[0]
+        return response
+        
     except Exception as e:
         print("Error in generate_response:")
         print(traceback.format_exc())  # Logs the full traceback