st0bb3n commited on
Commit
8da78db
1 Parent(s): 2df3b4c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,8 +4,8 @@ from datasets import load_dataset
4
  import torch
5
 
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')
 
4
  import torch
5
 
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')