Update app.py

app.py

@@ -35,7 +35,7 @@ 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(f"{user_prompt}", "").strip()
   assistant_response = assistant_response.replace("You are a helpful AI assistant.", "").strip()
   
   return assistant_response