Hugging Face's logo

Spaces:
MagnaSC
/
ImgSearch
Running on Zero

App Files Community
AkinyemiAra commited on
Commit
43a18b1
·
verified ·
1 Parent(s): 252afbf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,8 +8,8 @@ import os
8
  import spaces
9
 
10
  # Load model/processor
11
- model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
12
- processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
13
  model.eval()
14
 
15
  DATASET_DIR = Path("dataset")
 
8
  import spaces
9
 
10
  # Load model/processor
11
+ model = CLIPModel.from_pretrained("openai/clip-vit-large-patch14")
12
+ processor = CLIPProcessor.from_pretrained("openai/clip-vit-large-patch14")
13
  model.eval()
14
 
15
  DATASET_DIR = Path("dataset")