Update app.py

app.py

@@ -128,11 +128,11 @@ def hermes_generate_response(msg_prompt: str) -> dict:
         <|im_start|>assistant
         '''
         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
-        
+      
+         # Separate user's prompt and assistant's response
+        response_lines = pipe_output.split('\n')
+        user_prompt = response_lines[0].strip()
+        assistant_response = response_lines[-1].strip()
         
         return {"user": msg_prompt, "assistant": assistant_response}
     except Exception as e: