Upload demo/infer/run_caption_bbox_infer.py with huggingface_hub

demo/infer/run_caption_bbox_infer.py

@@ -21,14 +21,21 @@ except ImportError:
     )
 
 PROJECT_ROOT = Path(__file__).resolve().parents[2]
-DEFAULT_MODEL_REPO_ID = "SynLayers/Bbox-caption-8b"
 
 
 def resolve_default_bbox_model() -> str:
-    env_path = os.environ.get("SYNLAYERS_BBOX_MODEL") or os.environ.get("SYNLAYERS_MODEL_REPO")
-    if env_path and Path(env_path).exists():
-        return str(Path(env_path))
-    return env_path or DEFAULT_MODEL_REPO_ID
+    env_model = os.environ.get("SYNLAYERS_BBOX_MODEL") or os.environ.get("SYNLAYERS_BBOX_MODEL_REPO")
+    if env_model:
+        return env_model
+
+    candidates = [
+        PROJECT_ROOT if (PROJECT_ROOT / "config.json").exists() and (PROJECT_ROOT / "tokenizer_config.json").exists() else None,
+        PROJECT_ROOT / "Bbox-caption-8b",
+    ]
+    for candidate in candidates:
+        if candidate and candidate.exists():
+            return str(candidate)
+    return "SynLayers/Bbox-caption-8b"
 
 
 CAPTION_BBOX_PROMPT_TOP_LEFT = (