MichaelMM2000 commited on
Commit
407eabd
·
verified ·
1 Parent(s): 928603e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from transformers import pipeline
3
 
4
  # Load models
5
- vit_classifier = pipeline("image-classification", model="jarinschnierl/vit-base-oxford-iiit-pets")
6
  clip_detector = pipeline(model="openai/clip-vit-large-patch14", task="zero-shot-image-classification")
7
 
8
  labels_oxford_pets = [
 
2
  from transformers import pipeline
3
 
4
  # Load models
5
+ vit_classifier = pipeline("image-classification", model="MichaelMM2000/vit-base-oxford-iiit-pets")
6
  clip_detector = pipeline(model="openai/clip-vit-large-patch14", task="zero-shot-image-classification")
7
 
8
  labels_oxford_pets = [