SynLayers commited on
Commit
33ebab7
·
verified ·
1 Parent(s): 1a12879

Upload demo/infer/run_caption_bbox_infer.py with huggingface_hub

Browse files
demo/infer/run_caption_bbox_infer.py CHANGED
@@ -24,7 +24,7 @@ PROJECT_ROOT = Path(__file__).resolve().parents[2]
24
 
25
 
26
  def resolve_default_bbox_model() -> str:
27
- env_path = os.environ.get("SYNLAYERS_BBOX_MODEL")
28
  candidates = [
29
  Path(env_path) if env_path else None,
30
  PROJECT_ROOT if (PROJECT_ROOT / "config.json").exists() and (PROJECT_ROOT / "tokenizer_config.json").exists() else None,
 
24
 
25
 
26
  def resolve_default_bbox_model() -> str:
27
+ env_path = os.environ.get("SYNLAYERS_BBOX_MODEL") or os.environ.get("SYNLAYERS_MODEL_REPO")
28
  candidates = [
29
  Path(env_path) if env_path else None,
30
  PROJECT_ROOT if (PROJECT_ROOT / "config.json").exists() and (PROJECT_ROOT / "tokenizer_config.json").exists() else None,