Update app.py

app.py

@@ -5,6 +5,7 @@ import numpy as np
 import cv2
 from PIL import Image
 import spaces
+import io
 
 # 🌟 Auto-detect device (CPU/GPU)
 device = "cuda"
@@ -57,8 +58,12 @@ def generate_image(prompt, input_image, low_threshold, high_threshold, strength,
         controlnet_conditioning_scale=float(controlnet_conditioning_scale),
         strength=strength
     ).images[0]
+
+    result_buffer = io.BytesIO()
+    result.save(result_buffer, format="JPEG")
+    result_buffer.seek(0)
     
-    return edge_detected, result
+    return edge_detected, result_buffer
 
 # 🖥️ Gradio UI
 with gr.Blocks() as demo: