fix

README.md

@@ -4,7 +4,7 @@ emoji: 🧞
 colorFrom: blue
 colorTo: purple
 sdk: gradio
-sdk_version: 4.37.2
+sdk_version: 4.44.1
 app_file: app.py
 pinned: true
 license: mit

app.py

@@ -4,7 +4,7 @@ import os
 is_spaces = True if os.environ.get('SPACE_ID') else False
 if is_spaces:
     import spaces
-from gradio_client import Client, handle_file
+from gradio_client import Client, handle_file, utils as gradio_client_utils
 from huggingface_hub import snapshot_download, HfApi
 import uuid
 import shutil
@@ -34,6 +34,21 @@ mage_vl_client = None
 caption_cache = {}
 
 
+# Gradio 4.x's API schema formatter assumes every JSON Schema node is a dict.
+# Pydantic can legally emit a boolean node such as `additionalProperties: false`,
+# which otherwise crashes the Space's root route before the UI is displayed.
+_original_schema_to_python_type = gradio_client_utils._json_schema_to_python_type
+
+
+def _safe_schema_to_python_type(schema, defs):
+    if isinstance(schema, bool):
+        return "Any"
+    return _original_schema_to_python_type(schema, defs)
+
+
+gradio_client_utils._json_schema_to_python_type = _safe_schema_to_python_type
+
+
 # A Space configured for ZeroGPU must register at least one GPU function during
 # startup. Captioning itself runs in the separate Mage-VL Space, so this probe is
 # intentionally not placed around run_captioning (which would waste local quota
@@ -1178,4 +1193,4 @@ If you prefer command line, you can run our [training script]({training_script_u
     #demo.load(fn=swap_opacity, outputs=[main_ui], queue=False, concurrency_limit=50)
 if __name__ == "__main__":
     demo.queue()
-    demo.launch(share=True)
+    demo.launch(share=not is_spaces)

requirements.txt

@@ -2,6 +2,6 @@ autotrain-advanced
 numpy==1.26.4
 huggingface-hub==0.27.0
 python-slugify
-gradio>=4.37,<6
-gradio_client>=1.0
+gradio==4.44.1
+gradio_client==1.3.0
 spaces