RobotJelly commited on
Commit
bb51e19
·
1 Parent(s): fb23d7b
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
17
  processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
18
  tokenizer = CLIPTokenizer.from_pretrained("openai/clip-vit-base-patch32")
19
 
20
- model = model.to(device)
21
 
22
  # taking photo IDs
23
  photo_ids = pd.read_csv("./photo_ids.csv")
 
17
  processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
18
  tokenizer = CLIPTokenizer.from_pretrained("openai/clip-vit-base-patch32")
19
 
20
+ model = model.to(device)
21
 
22
  # taking photo IDs
23
  photo_ids = pd.read_csv("./photo_ids.csv")