lindritdev commited on
Commit
8cffda4
·
verified ·
1 Parent(s): d6b4384

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="lindritdev/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="lindritdev/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 = [