nkanungo commited on
Commit
f8791f7
·
verified ·
1 Parent(s): 3e5f0e7

Update app.py

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