IvaElen commited on
Commit
a3426b0
1 Parent(s): 1cd1cac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,9 +15,9 @@ from get_similiarty import get_similiarity
15
  device = "cuda" if torch.cuda.is_available() else "cpu"
16
 
17
  #load model -resnet50
18
- model_resnet = torch.load("model.pt", device)
19
  #load model - ViT-B/32
20
- model_vit = torch.load("model_vit.pt", device)
21
 
22
  #Распаковка ZIP-файла с фотографиями
23
  zip_file_path = "sample.zip"
 
15
  device = "cuda" if torch.cuda.is_available() else "cpu"
16
 
17
  #load model -resnet50
18
+ model_resnet, prerocess = clip.load("RN50", device=device)
19
  #load model - ViT-B/32
20
+ model_vit, preprocess = clip.load('ViT-B/32', device)
21
 
22
  #Распаковка ZIP-файла с фотографиями
23
  zip_file_path = "sample.zip"