Update app.py

app.py

@@ -61,10 +61,11 @@ def generate_response(prompt: str) -> str:
         top_k=40,
         repetition_penalty=1.1
     )
-    pipe(prompt_template + user_prompt)
+    # Generate the response
+    response = pipe(prompt_template + user_prompt)
+    generated_text = response[0]['generated_text']
 
-                
-    return {"user": prompt, "assistant": [0]['generated_text']}
+    return {"user": prompt, "assistant": generated_text
 
   
 def generate_prompt_response(persona_prompt: str, prompt: str) -> dict: