Update app.py

app.py

@@ -27,7 +27,8 @@ def generate(text):
         {'role': 'system', 'content': system_prompt},
         {'role': 'user', 'content': text}
     ]
-    return pipe(messages, **generate_kwargs)['generated_text'][-1]['content']
+    response = pipe(messages, **generate_kwargs)[0]['generated_text']
+    return response[-1]['content']
 
     
 if __name__ == "__main__":