Fix Gradio schema bug: Interface+filepath, gradio-client>=1.4

README.md

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

app.py

@@ -3,7 +3,10 @@
 from __future__ import annotations
 
 import os
-from pathlib import Path
+
+# Hugging Face Spaces: required before importing gradio
+os.environ.setdefault("GRADIO_SERVER_NAME", "0.0.0.0")
+os.environ.setdefault("GRADIO_SERVER_PORT", os.environ.get("PORT", "7860"))
 
 import gradio as gr
 import numpy as np
@@ -11,33 +14,18 @@ from PIL import Image
 
 from prebackbone_infer import enrich_pair, get_enricher
 
-EXAMPLES_DIR = Path(__file__).resolve().parent / "examples"
 TITLE = "RefDiffNet — PCB Reference–Defect Enrichment"
 DESCRIPTION = """
 Upload a **defect PCB image** and its **golden reference** (same board, no defect).
-The trained A11_CA prebackbone fuses them into an **enriched** image used by YOLO detection.
-
-Formula: `enriched = defect + α · gate · delta`
+The A11_CA prebackbone outputs an **enriched** image: `enriched = defect + α · gate · delta`
 """
 
 
-def _examples() -> list[list[str]]:
-    if not EXAMPLES_DIR.exists():
-        return []
-    inputs = sorted(EXAMPLES_DIR.glob("*_input.jpg"))
-    out: list[list[str]] = []
-    for inp in inputs:
-        ref = EXAMPLES_DIR / inp.name.replace("_input.", "_reference.")
-        if ref.exists():
-            out.append([str(inp), str(ref)])
-    return out
-
-
-def run_demo(defect_img, reference_img):
-    if defect_img is None or reference_img is None:
+def run_demo(defect_path: str | None, reference_path: str | None):
+    if not defect_path or not reference_path:
         raise gr.Error("Please upload both defect and reference images.")
 
-    defect_rgb, ref_rgb, enriched_rgb = enrich_pair(defect_img, reference_img)
+    defect_rgb, ref_rgb, enriched_rgb = enrich_pair(defect_path, reference_path)
 
     h = max(defect_rgb.shape[0], ref_rgb.shape[0], enriched_rgb.shape[0])
 
@@ -54,33 +42,29 @@ def run_demo(defect_img, reference_img):
 def _preload_model():
     try:
         get_enricher()
+        print("[RefDiffNet] Prebackbone loaded.")
     except FileNotFoundError as e:
-        print(f"[warn] Model not loaded yet: {e}")
+        print(f"[RefDiffNet] WARN: {e}")
 
 
 if __name__ == "__main__":
     _preload_model()
 
-    with gr.Blocks(title=TITLE) as demo:
-        gr.Markdown(f"# {TITLE}")
-        gr.Markdown(DESCRIPTION)
-
-        with gr.Row():
-            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="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])
-
-        ex = _examples()
-        if ex:
-            gr.Examples(examples=ex, inputs=[defect_in, ref_in], label="Example pairs")
+    demo = gr.Interface(
+        fn=run_demo,
+        inputs=[
+            gr.Image(type="filepath", label="Defect image (input)"),
+            gr.Image(type="filepath", label="Golden reference"),
+        ],
+        outputs=[
+            gr.Image(type="pil", label="Enriched output"),
+            gr.Image(type="pil", label="Defect | Reference | Enriched"),
+        ],
+        title=TITLE,
+        description=DESCRIPTION,
+        allow_flagging="never",
+    )
 
-    # 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")),

setup.sh

@@ -4,7 +4,10 @@ set -euo pipefail
 cd "$(dirname "$0")"
 # 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"
+# Fix Gradio 4.44.0 + pydantic schema bug (TypeError: bool is not iterable)
+pip install -q "gradio-client>=1.4.0,<2.0.0"
 export YOLO_CONFIG_DIR="${YOLO_CONFIG_DIR:-/tmp/Ultralytics}"
+export GRADIO_SERVER_NAME="${GRADIO_SERVER_NAME:-0.0.0.0}"
 mkdir -p "${YOLO_CONFIG_DIR}"
 
 if [[ -d vendor ]]; then