fix4: torch.load

app.py

@@ -82,6 +82,14 @@ from pathlib import Path
 
 import gradio as gr
 import torch
+
+# ── Patch torch.load for PyTorch 2.7 (weights_only=True default breaks YOLO) ─
+_orig_tload = torch.load
+def _safe_tload(f, map_location=None, pickle_module=None,
+                weights_only=None, mmap=None, **kwargs):
+    return _orig_tload(f, map_location=map_location, weights_only=False, **kwargs)
+torch.load = _safe_tload
+
 import huggingface_hub   # noqa: F401  (ensure installed; used in _hf_ckpt)
 
 # ─────────────────────────────────────────────────────────────────────────────
@@ -1532,7 +1540,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-class</li>
+    <li>Coronary segmentation — YOLOv8m-seg (26-class)</li>
     <li>Binary vessel mask — ResUNet</li>
     <li>FFR estimation — QFR v4</li>
     <li>SYNTAX score computation</li>
@@ -1584,7 +1592,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 (YOLOv8m-seg)</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>')