Spaces:
Runtime error
Runtime error
Commit ·
85db738
1
Parent(s): e86b729
fixed error
Browse files
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
|