Update app.py

app.py

@@ -148,7 +148,7 @@ def main(query):
 
     response = generator(formatted_prompt, return_full_text=False)
 
-    return response
+    return response[0]['generated_text']
 
 with gr.Blocks() as demo:
     #gr.Image("../Documentation/Context Diagram.png", scale=2)