methodw commited on
Commit
c0c77aa
·
1 Parent(s): eaf7419

get model from hf

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,8 +9,8 @@ import faiss
9
 
10
  # Init similarity search AI model and processor
11
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
12
- processor = AutoImageProcessor.from_pretrained("./dinov2-large")
13
- model = AutoModel.from_pretrained("./dinov2-large")
14
  model.config.return_dict = False # Set return_dict to False for JIT tracing
15
  model.to(device)
16
 
 
9
 
10
  # Init similarity search AI model and processor
11
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
12
+ processor = AutoImageProcessor.from_pretrained("facebook/dinov2-large")
13
+ model = AutoModel.from_pretrained("facebook/dinov2-large")
14
  model.config.return_dict = False # Set return_dict to False for JIT tracing
15
  model.to(device)
16