mjwong commited on
Commit
ff692cc
·
verified ·
1 Parent(s): 5aa818b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -57,13 +57,13 @@ logger.info("Country code mappings loaded successfully.")
57
  # Load BioCLIP Model from Hugging Face
58
  logger.info("Loading model from Hugging Face...")
59
  repo_dir = snapshot_download(MODEL_NAME, revision=MODEL_REVISION)
 
60
  cfg = Path(repo_dir, "open_clip_config.json")
61
  wts = Path(repo_dir, "open_clip_pytorch_model.bin")
62
  assert cfg.exists(), f"Missing config at {cfg}"
63
  assert wts.exists(), f"Missing weights at {wts}"
64
  model, _, preprocess = create_model_and_transforms(
65
- model_name=f"local-dir:{repo_dir}",
66
- cache_dir=None
67
  )
68
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
69
  tokenizer = get_tokenizer(MODEL_NAME)
 
57
  # Load BioCLIP Model from Hugging Face
58
  logger.info("Loading model from Hugging Face...")
59
  repo_dir = snapshot_download(MODEL_NAME, revision=MODEL_REVISION)
60
+ repo_dir = str(Path(repo_dir).resolve())
61
  cfg = Path(repo_dir, "open_clip_config.json")
62
  wts = Path(repo_dir, "open_clip_pytorch_model.bin")
63
  assert cfg.exists(), f"Missing config at {cfg}"
64
  assert wts.exists(), f"Missing weights at {wts}"
65
  model, _, preprocess = create_model_and_transforms(
66
+ model_name=f"local-dir:{repo_dir}"
 
67
  )
68
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
69
  tokenizer = get_tokenizer(MODEL_NAME)