Update app.py
Browse files
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)
|