Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,8 @@ import torch
|
|
6 |
# Load the model and processor
|
7 |
model_id = "DGurgurov/clip-vit-base-patch32-oxford-pets"
|
8 |
model = CLIPModel.from_pretrained(model_id)
|
9 |
-
|
|
|
10 |
|
11 |
# Define the inference function
|
12 |
def predict(image):
|
|
|
6 |
# Load the model and processor
|
7 |
model_id = "DGurgurov/clip-vit-base-patch32-oxford-pets"
|
8 |
model = CLIPModel.from_pretrained(model_id)
|
9 |
+
processor_id = "openai/clip-vit-base-patch32"
|
10 |
+
processor = CLIPProcessor.from_pretrained(processor_id)
|
11 |
|
12 |
# Define the inference function
|
13 |
def predict(image):
|