Update app.py

app.py

@@ -35,7 +35,9 @@ def generate_text(user_prompt):
 
   # Extract the assistant's response (assuming it starts with "<|assistant|>")
   assistant_response = generated_text.split("<|assistant|>")[-1]
-
+  assistant_response = assistant_response.replace(f"{prompt}", "").strip()
+  assistant_response = assistant_response.replace("You are a helpful AI assistant.", "").strip()
+  
   return assistant_response
 
 # Define the Gradio interface