Hugging Face's logo

Spaces:
ClassCat
/
ViT-ImageNet-Classification
Runtime error

App Files Community
ClassCat commited on
Commit
1f42eb9
•
1 Parent(s): 78911a4

update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -9,6 +9,8 @@ from PIL import Image
9
  feature_extractor = ViTFeatureExtractor.from_pretrained('google/vit-base-patch16-224')
10
  model = ViTForImageClassification.from_pretrained('google/vit-base-patch16-224')
11
 
 
 
12
  examples_dir = './samples'
13
  example_files = glob.glob(os.path.join(examples_dir, '*.jpg'))
14
 
 
9
  feature_extractor = ViTFeatureExtractor.from_pretrained('google/vit-base-patch16-224')
10
  model = ViTForImageClassification.from_pretrained('google/vit-base-patch16-224')
11
 
12
+ import os, glob
13
+
14
  examples_dir = './samples'
15
  example_files = glob.glob(os.path.join(examples_dir, '*.jpg'))
16