Update app.py

app.py

@@ -64,10 +64,9 @@ def generate_response(prompt: str) -> str:
     # Generate the response
     generated_response = pipe(prompt_template)[0]['generated_text']
 
-    # Extract the assistant's reply
-    assistant_reply = generated_response.split('\n')[1]
+    assistant_reply = generated_response.split('\n\n')[1]
 
-    return {"user": prompt, "assistant": generated_response}
+    return {"user": prompt, "assistant": assistant_reply}
 
   
 def generate_prompt_response(persona_prompt: str, prompt: str) -> dict: