Update app.py

app.py

@@ -13,7 +13,7 @@ async def generate_image(prompt):
         # Generate an image based on the prompt
         output = await asyncio.to_thread(model, prompt=prompt, num_inference_steps=1, guidance_scale=0.0)
         print(f"Model output: {output}")
-
+        return output
         # Check if the model returned images
         if isinstance(output.images, list) and len(output.images) > 0:
             image = output.images[0]