Fix Gradio 4.49 pil images and HF launch (no share)

README.md

@@ -4,7 +4,7 @@ emoji: 🔬
 colorFrom: blue
 colorTo: green
 sdk: gradio
-sdk_version: "4.44.0"
+sdk_version: "4.49.1"
 python_version: "3.10"
 app_file: app.py
 pinned: false

app.py

@@ -39,7 +39,6 @@ def run_demo(defect_img, reference_img):
 
     defect_rgb, ref_rgb, enriched_rgb = enrich_pair(defect_img, reference_img)
 
-    # Side-by-side comparison for the demo panel
     h = max(defect_rgb.shape[0], ref_rgb.shape[0], enriched_rgb.shape[0])
 
     def _pad(im: np.ndarray) -> np.ndarray:
@@ -49,7 +48,7 @@ def run_demo(defect_img, reference_img):
         return np.pad(im, ((0, pad), (0, 0), (0, 0)), mode="constant", constant_values=114)
 
     row = np.concatenate([_pad(defect_rgb), _pad(ref_rgb), _pad(enriched_rgb)], axis=1)
-    return enriched_rgb, row
+    return Image.fromarray(enriched_rgb), Image.fromarray(row)
 
 
 def _preload_model():
@@ -67,17 +66,13 @@ if __name__ == "__main__":
         gr.Markdown(DESCRIPTION)
 
         with gr.Row():
-            defect_in = gr.Image(label="Defect image (input)", type="numpy", image_mode="RGB")
-            ref_in = gr.Image(label="Golden reference", type="numpy", image_mode="RGB")
+            defect_in = gr.Image(label="Defect image (input)", type="pil", image_mode="RGB")
+            ref_in = gr.Image(label="Golden reference", type="pil", image_mode="RGB")
 
         run_btn = gr.Button("Generate enriched image", variant="primary")
         with gr.Row():
-            enriched_out = gr.Image(label="Enriched output", type="numpy", image_mode="RGB")
-            compare_out = gr.Image(
-                label="Defect | Reference | Enriched",
-                type="numpy",
-                image_mode="RGB",
-            )
+            enriched_out = gr.Image(label="Enriched output", type="pil", image_mode="RGB")
+            compare_out = gr.Image(label="Defect | Reference | Enriched", type="pil", image_mode="RGB")
 
         run_btn.click(fn=run_demo, inputs=[defect_in, ref_in], outputs=[enriched_out, compare_out])
 
@@ -85,8 +80,10 @@ if __name__ == "__main__":
         if ex:
             gr.Examples(examples=ex, inputs=[defect_in, ref_in], label="Example pairs")
 
+    # HF Spaces: bind 0.0.0.0, never use share=True
     demo.launch(
         server_name="0.0.0.0",
         server_port=int(os.environ.get("PORT", "7860")),
-        share=os.environ.get("GRADIO_SHARE", "").lower() in {"1", "true", "yes"},
+        share=False,
+        show_error=True,
     )

requirements.txt

@@ -1,4 +1,5 @@
-gradio>=4.44.0
+gradio>=4.49.1,<5.0.0
+gradio-client>=1.3.0,<1.6.0
 # Python 3.13+ removed stdlib audioop; pydub (gradio dep) needs this backport
 audioop-lts>=0.2.1; python_version >= "3.13"
 torch>=2.0.0

setup.sh

@@ -2,8 +2,10 @@
 # Hugging Face Spaces: install vendored custom Ultralytics before app starts.
 set -euo pipefail
 cd "$(dirname "$0")"
-# HF base image may install huggingface_hub 1.x; Gradio 4.44 needs <1.0 (HfFolder)
+# HF base image may install huggingface_hub 1.x; Gradio 4.x needs <1.0 (HfFolder)
 pip install -q "huggingface_hub>=0.23.0,<1.0"
+export YOLO_CONFIG_DIR="${YOLO_CONFIG_DIR:-/tmp/Ultralytics}"
+mkdir -p "${YOLO_CONFIG_DIR}"
 
 if [[ -d vendor ]]; then
   pip install -q -e ./vendor