methodw commited on
Commit
64467e5
·
1 Parent(s): 5aee97b

force download model

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -14,7 +14,9 @@ load_dotenv()
14
 
15
  # Init similarity search AI model and processor
16
  torch_device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
17
- dino_v2_model = AutoModel.from_pretrained("facebook/dinov2-base").to(torch_device)
 
 
18
  dino_v2_image_processor = AutoImageProcessor.from_pretrained("facebook/dinov2-base")
19
 
20
  # MongoDB
 
14
 
15
  # Init similarity search AI model and processor
16
  torch_device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
17
+ dino_v2_model = AutoModel.from_pretrained(
18
+ "facebook/dinov2-base", force_download=True
19
+ ).to(torch_device)
20
  dino_v2_image_processor = AutoImageProcessor.from_pretrained("facebook/dinov2-base")
21
 
22
  # MongoDB