anas-gouda commited on
Commit
3c8be8e
·
1 Parent(s): 91b36d4

load classification model from the repo

Browse files
Files changed (1) hide show
  1. 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,