Update app.py

app.py

@@ -240,14 +240,12 @@ async def generate_image(
 
         # Call existing inference function with the provided parameters
         generated_image, _, _, _ = inference(control_image, prompt, "", guidance_scale, controlnet_scale, 0, controlnet_end, upscaler_strength, seed, sampler_type)
+        if generated_image is None:
+            return "Failed to generate image"
 
-        # Convert the PIL Image to bytes
-        img_bytes = io.BytesIO()
-        generated_image.save(img_bytes, format='PNG')
-        img_bytes.seek(0)
 
-        # Return the generated image as response
-        return FileResponse(img_bytes, media_type='image/png', headers={'Content-Disposition': 'inline; filename=generated_image.png'})
+        logger.debug("Generated Image Size: %s", generated_image.size)
+        return generated_image
 
     except Exception as e:
         # Handle exceptions and return an error message if something goes wrong