Update app.py

app.py

@@ -129,8 +129,12 @@ def hermes_generate_response(msg_prompt: str) -> dict:
         '''
         pipe_output = pipe(prompt_template)[0]['generated_text']
         #assistant_reply = pipe_output.split('\n\n')[1]
+        # Separate user's prompt and assistant's response
+        response_lines = pipe_output.split('\n\n')
+        assistant_response = response_lines[-1].strip()  # Get the last line
         
-        return {"user": msg_prompt, "assistant": pipe_output}
+        
+        return {"user": msg_prompt, "assistant": assistant_response}
     except Exception as e:
         return {"error": str(e)}