Update app.py

app.py

@@ -19,11 +19,11 @@ from transformers import pipeline
 # ============================
 # Config
 # ============================
-#DEFAULT_MODEL_ID = "facebook/dinov3-vits16plus-pretrain-lvd1689m"
-#ALT_MODEL_ID = "facebook/dinov3-vith16plus-pretrain-lvd1689m"
+DEFAULT_MODEL_ID = "facebook/dinov3-vits16plus-pretrain-lvd1689m"
+ALT_MODEL_ID = "facebook/dinov3-vith16plus-pretrain-lvd1689m"
 
-DEFAULT_MODEL_ID = "onnx-community/dinov3-vits16-pretrain-lvd1689m-ONNX"
-ALT_MODEL_ID = "onnx-community/dinov3-vith16-pretrain-lvd1689m-ONNX"
+#DEFAULT_MODEL_ID = "onnx-community/dinov3-vits16-pretrain-lvd1689m-ONNX"
+#ALT_MODEL_ID = "onnx-community/dinov3-vith16-pretrain-lvd1689m-ONNX"
 
 AVAILABLE_MODELS = [DEFAULT_MODEL_ID, ALT_MODEL_ID]
 
@@ -60,7 +60,7 @@ def load_model_from_hubold(model_id: str):
     return mdl
 
     
-def load_model_from_hub(model_id: str):
+def load_model_from_hubold2(model_id: str):
     print(f"Loading model '{model_id}' from HF Hub…")
     token = os.environ.get("HF_TOKEN")
     # Use pipeline instead of AutoModel
@@ -74,6 +74,17 @@ def load_model_from_hub(model_id: str):
     print(f"✅ Loaded '{model_id}' on {DEVICE}")
     return extractor
 
+def load_model_from_hub(model_id: str):
+    print(f"Loading model '{model_id}' from HF Hub…")
+    token = os.environ.get("HF_TOKEN")
+    mdl = AutoModel.from_pretrained(
+        model_id,
+        token=token,
+        trust_remote_code=True,
+    )
+    mdl.to(DEVICE).eval()
+    print(f"✅ Loaded '{model_id}' on {DEVICE}")
+    return mdl