Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
3c8be8e
1
Parent(s):
91b36d4
load classification model from the repo
Browse files- utils/models.py +1 -0
utils/models.py
CHANGED
|
@@ -38,6 +38,7 @@ def load_dounseen_model(
|
|
| 38 |
device: torch.device,
|
| 39 |
) -> UnseenClassifier:
|
| 40 |
unseen_classifier = UnseenClassifier(
|
|
|
|
| 41 |
gallery_images=None,
|
| 42 |
gallery_buffered_path=None,
|
| 43 |
augment_gallery=False,
|
|
|
|
| 38 |
device: torch.device,
|
| 39 |
) -> UnseenClassifier:
|
| 40 |
unseen_classifier = UnseenClassifier(
|
| 41 |
+
classification_model_path="models/dounseen/vit_b_16_epoch_199_augment.pth",
|
| 42 |
gallery_images=None,
|
| 43 |
gallery_buffered_path=None,
|
| 44 |
augment_gallery=False,
|