Update app.py

app.py

@@ -9,7 +9,7 @@ def predict(query):
     messages = []
     messages.append({"role": "system","content": "you are a helpful assistant"})
     messages.append({"role": "user","content": query})
-    return pipe(messages,max_new_tokens=256)[0]["generated_text"][-1]
+    return pipe(messages,max_new_tokens=256)[0]["generated_text"][-1]['content']
 
 gradio_app = gr.Interface(
     predict,