Update app.py

app.py

@@ -17,7 +17,6 @@ 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()
@@ -68,8 +67,7 @@ gradio_interface = gr.Interface(
     ],
     outputs="json",
     concurrency_limit=20  # Set a high concurrency limit
-).queue(default_concurrency_limit=20) 
+).queue(default_concurrency_limit=20)
 
 if __name__ == "__main__":
     gradio_interface.launch()  # No need for share=True for local testing
-