Updated prediction format

app.py

@@ -25,7 +25,7 @@ def generate(text):
         temperature=temperature
     )
 
-    return [i["generated_text"] for i in generated]
+    return generated[0]["generated_text"]
 
 
 iface = gr.Interface(fn=generate, inputs="text", outputs="text")