adisaljusi commited on
Commit
816b68d
·
1 Parent(s): 09f9cfa

Update model name and format label list for clarity

Browse files
Files changed (1) hide show
  1. app.py +11 -3
app.py CHANGED
@@ -14,15 +14,23 @@ OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
14
  openai_client = OpenAI(api_key=OPENAI_API_KEY) if OPENAI_API_KEY else None
15
 
16
  # Load models
17
- vit_classifier = pipeline("image-classification", model="adisaljusi/vit-base-cifar10")
18
  clip_detector = pipeline(
19
  model="openai/clip-vit-large-patch14",
20
  task="zero-shot-image-classification",
21
  )
22
 
23
  labels_cifar10 = [
24
- "airplane", "automobile", "bird", "cat", "deer",
25
- "dog", "frog", "horse", "ship", "truck",
 
 
 
 
 
 
 
 
26
  ]
27
 
28
 
 
14
  openai_client = OpenAI(api_key=OPENAI_API_KEY) if OPENAI_API_KEY else None
15
 
16
  # Load models
17
+ vit_classifier = pipeline("image-classification", model="adisaljusi/cifar10-vit")
18
  clip_detector = pipeline(
19
  model="openai/clip-vit-large-patch14",
20
  task="zero-shot-image-classification",
21
  )
22
 
23
  labels_cifar10 = [
24
+ "airplane",
25
+ "automobile",
26
+ "bird",
27
+ "cat",
28
+ "deer",
29
+ "dog",
30
+ "frog",
31
+ "horse",
32
+ "ship",
33
+ "truck",
34
  ]
35
 
36