nkanungo commited on
Commit
a7eca3a
·
verified ·
1 Parent(s): eb0cd69

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ from sentence_transformers import SentenceTransformer, util
23
  device = "cuda" if torch.cuda.is_available() else "cpu"
24
 
25
  # Define model
26
- model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32").to(device)
27
  processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
28
  tokenizer = CLIPTokenizer.from_pretrained("openai/clip-vit-base-patch32")
29
 
 
23
  device = "cuda" if torch.cuda.is_available() else "cpu"
24
 
25
  # Define model
26
+ model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32", config=config.vision_config).to(device)
27
  processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
28
  tokenizer = CLIPTokenizer.from_pretrained("openai/clip-vit-base-patch32")
29