fix: torch.load

app.py

@@ -24,6 +24,30 @@ if "audioop" not in sys.modules:
 if "pyaudioop" not in sys.modules:
     sys.modules["pyaudioop"] = types.ModuleType("pyaudioop")
 
+# ── Patch gradio_client _json_schema_to_python_type (APIInfoParseError fix) ──
+def _patch_gradio_client():
+    try:
+        import gradio_client.utils as _gcu
+        _orig = _gcu._json_schema_to_python_type
+        def _safe(schema, defs=None):
+            if not isinstance(schema, dict):
+                return "Any"
+            try:
+                return _orig(schema, defs)
+            except Exception:
+                return "Any"
+        _gcu._json_schema_to_python_type = _safe
+        _orig_top = _gcu.json_schema_to_python_type
+        def _safe_top(schema, defs=None):
+            try:
+                return _orig_top(schema, defs)
+            except Exception:
+                return "Any"
+        _gcu.json_schema_to_python_type = _safe_top
+    except Exception:
+        pass
+_patch_gradio_client()
+
 import os
 import sys
 import json
@@ -701,11 +725,9 @@ def _load_yolo():
             f"Check HF_TOKEN secret and repo MuhammadAdil63/angio-ai-checkpoints"
         )
     print(f"[YOLO] Loading checkpoint: {ckpt}")
-    # PyTorch 2.6+ requires explicit allowlisting of custom globals
     try:
         import torch.serialization
         from ultralytics.nn.tasks import SegmentationModel
-        from ultralytics.nn.modules import (Conv, C2f, SPPF, Detect, Segment)
         torch.serialization.add_safe_globals([SegmentationModel])
     except Exception as _e:
         print(f"[YOLO] safe_globals warning: {_e}")
@@ -1494,7 +1516,7 @@ with gr.Blocks(css=CSS, title="Angio AI") as demo:
   <ol style="font-size:13px;color:#1a2533;line-height:1.9;padding-left:16px">
     <li>Best frame extracted from video</li>
     <li>Stenosis detection — Mask2Former</li>
-    <li>Coronary segmentation — (26-classes)</li>
+    <li>Coronary segmentation — 26-classes</li>
     <li>Binary vessel mask — ResUNet</li>
     <li>FFR estimation — QFR v4</li>
     <li>SYNTAX score computation</li>
@@ -1546,7 +1568,7 @@ with gr.Blocks(css=CSS, title="Angio AI") as demo:
 
             with gr.Row():
                 with gr.Column():
-                    gr.HTML('<div class="card-title" style="margin-bottom:6px">Coronary segmentation</div>')
+                    gr.HTML('<div class="card-title" style="margin-bottom:6px">Coronary segmentation </div>')
                     seg_out   = gr.Image(label="26-class overlay", height=300)
                 with gr.Column():
                     gr.HTML('<div class="card-title" style="margin-bottom:6px">FFR analysis</div>')
@@ -1608,5 +1630,5 @@ if __name__ == "__main__":
     demo.launch(
         server_name="0.0.0.0",
         server_port=7860,
-        share=False,
+        share=True,
     )