Spaces:

DGurgurov
/

clip-pets-classifier

App Files Files Community

DGurgurov commited on Jun 17

Commit

103d031

•

1 Parent(s): 5bd2a28

Update app.py

Files changed (1) hide show

app.py +2 -1

app.py CHANGED Viewed

@@ -6,7 +6,8 @@ import torch
 # Load the model and processor
 model_id = "DGurgurov/clip-vit-base-patch32-oxford-pets"
 model = CLIPModel.from_pretrained(model_id)
-processor = CLIPProcessor.from_pretrained(model_id)
 # Define the inference function
 def predict(image):

 # Load the model and processor
 model_id = "DGurgurov/clip-vit-base-patch32-oxford-pets"
 model = CLIPModel.from_pretrained(model_id)
+processor_id = "openai/clip-vit-base-patch32"
+processor = CLIPProcessor.from_pretrained(processor_id)
 # Define the inference function
 def predict(image):