Upload folder using huggingface_hub

app.py

@@ -16,6 +16,32 @@ import gradio as gr
 import matplotlib.pyplot as plt
 from matplotlib.patches import Polygon
 
+
+def _patch_gradio_schema_bool_bug() -> None:
+    """
+    Work around Gradio/gradio_client schema parsing bug where boolean schema
+    values can crash API info generation.
+    """
+    try:
+        import gradio_client.utils as gc_utils  # type: ignore
+    except Exception:
+        return
+
+    if getattr(gc_utils.get_type, "__name__", "") == "_safe_get_type":
+        return
+
+    original_get_type = gc_utils.get_type
+
+    def _safe_get_type(schema):
+        if isinstance(schema, bool):
+            return "boolean"
+        return original_get_type(schema)
+
+    gc_utils.get_type = _safe_get_type
+
+
+_patch_gradio_schema_bool_bug()
+
 try:
     from src.data_collection import DataCollectionManager, classify_from_percentages_simple
 except ImportError:
@@ -1007,8 +1033,15 @@ if __name__ == "__main__":
         attention_reduction=args.attention_reduction,
         task_attention=args.task_attention,
     )
-    demo.launch(
-        server_name=args.server_name,
-        server_port=args.server_port,
-        share=args.share
-    )
+    is_space_runtime = bool(os.getenv("SPACE_ID") or os.getenv("HF_SPACE_ID"))
+    launch_kwargs = {
+        "server_name": args.server_name,
+        "server_port": args.server_port,
+        "share": args.share,
+    }
+    if is_space_runtime:
+        # Space does not need public-share tunnel and this avoids extra startup issues.
+        launch_kwargs["share"] = False
+        launch_kwargs["show_api"] = False
+
+    demo.launch(**launch_kwargs)

requirements.txt

@@ -12,7 +12,10 @@ opencv-python-headless>=4.8.0
 matplotlib>=3.7.0
 
 # WebUI - Required for Gradio interface
-gradio==4.44.0
+gradio==4.44.1
 
 # Hub sync/export (Space -> Dataset)
 huggingface_hub>=0.26.0,<1.0.0
+
+# Keep schema generation compatible with Gradio 4.x in Space runtime
+pydantic==2.10.6