st0bb3n commited on
Commit
147cf50
1 Parent(s): 8da78db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import torch
6
 
7
  #dataset = load_dataset("cifar100")
8
  #image = dataset["train"]["fine_label"]
9
- print("load and train dataset \n")
10
 
11
  feature_extractor = ViTFeatureExtractor.from_pretrained('google/vit-base-patch16-224')
12
  print("feature extractor \n")
 
6
 
7
  #dataset = load_dataset("cifar100")
8
  #image = dataset["train"]["fine_label"]
9
+ #print("load and train dataset \n")
10
 
11
  feature_extractor = ViTFeatureExtractor.from_pretrained('google/vit-base-patch16-224')
12
  print("feature extractor \n")