Update app.py

app.py

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