Update app.py

app.py

@@ -18,7 +18,7 @@ def query(inputs, parameters=None, ENDPOINT_URL=API_URL):
 #Defining text generation function
 def generate_text(inputs):
     output = query(inputs)
-    return output
+    return output[0]['generated_text']
 
 
 #Gradio Interface