Update app.py

app.py

@@ -235,13 +235,17 @@ async def generate_image(
         control_image = Image.open(temp_image_path)
 
         # 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)
+        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"
 
-
-        logger.debug("Generated Image Size: %s", generated_image.size)
-        return generated_image
+        # Save the generated image as binary data
+        output_image_io = io.BytesIO()
+        generated_image.save(output_image_io, format="PNG")
+        output_image_io.seek(0)
+        
+        # Return the image as a streaming response
+        return StreamingResponse(content=output_image_io, media_type="image/png")
 
     except Exception as e:
         # Handle exceptions and return an error message if something goes wrong