kautilya286 commited on
Commit
85db738
·
1 Parent(s): e86b729

fixed error

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
19
  print(f"[INFO] Using device: {device}")
20
 
21
  # Load the CLIP model and processor
22
- model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32").to(device)
23
  processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
24
 
25
  # Load the ensemble classifier model
 
19
  print(f"[INFO] Using device: {device}")
20
 
21
  # Load the CLIP model and processor
22
+ model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32", from_tf=True).to(device)
23
  processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
24
 
25
  # Load the ensemble classifier model