don't subscript return

app.py

@@ -42,7 +42,7 @@ def infer(prompt, negative_prompt, image):
         num_inference_steps=20
             )
 
-    return output.images[0]
+    return output.images
 
 gr.Interface(
     infer,