Straueri
/

vit-base-oxford-iiit-pets

Image Classification

Generated from Trainer

Model card Files Files and versions

Straueri commited on Apr 16, 2025

Commit

4c298ba

·

verified ·

1 Parent(s): 685dab6

Update app.py

Files changed (1) hide show

app.py +1 -1

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import pipeline
-classifier = pipeline("image-classification", model="kuhs/vit-base-oxford-iiit-pets")
 def classify_pet(image):
     results = classifier(image)

 import gradio as gr
 from transformers import pipeline
+classifier = pipeline("image-classification", model="Straueri/vit-base-oxford-iiit-pets")
 def classify_pet(image):
     results = classifier(image)