Update app.py

app.py

@@ -63,10 +63,10 @@ def generate_response(prompt: str) -> str:
         top_k=40,
         repetition_penalty=1.1
     )
-        generated_text = pipe(prompt_template)[0]['generated_text']
+    generated_text = pipe(prompt_template)[0]['generated_text']
         
         # Remove the "ASSISTANT:" prefix from the generated text
-        assistant_response = generated_text.replace("ASSISTANT:", "").strip()
+    assistant_response = generated_text.replace("ASSISTANT:", "").strip()
             
     return {"user": prompt, "assistant": assistant_response}