Demo: v2 head -> weights/head.onnx

app.py

@@ -137,7 +137,7 @@ def _ensure_v2():
         dl = lambda f: hf_hub_download(V2_PRIV, f, repo_type="model", token=TOKEN)
         _V2["img"] = _ort(dl("weights/image_encoder.onnx"))    # pixel_values -> pooled patches [1,256,1152]
         _V2["txt"] = _ort(dl("weights/text_encoder.onnx"))     # input_ids/mask -> L2 label embeds [N,1024]
-        _V2["head"] = _ort(dl("weights/head-v3-ep6.onnx"))     # patches + label_embeds -> probs
+        _V2["head"] = _ort(dl("weights/head.onnx"))            # patches + label_embeds -> probs
         _V2["ip"] = AutoImageProcessor.from_pretrained(V2_PRIV, subfolder="encoder/image", token=TOKEN)
         _V2["tk"] = AutoTokenizer.from_pretrained(V2_PRIV, subfolder="encoder/text", token=TOKEN)
     except Exception as e:  # noqa: BLE001