Update app.py

app.py

@@ -17,6 +17,7 @@ async def generate_image(prompt):
         # Check if the model returned images
         if isinstance(output.images, list) and len(output.images) > 0:
             image = output.images[0]
+            gradio.Imageimage
             buffered = BytesIO()
             image.save(buffered, format="JPEG")
             image_bytes = buffered.getvalue()