Update app.py

app.py

@@ -8,11 +8,12 @@ dataset = load_dataset("cifar100")
 image = dataset["train"]["fine_label"]
 print("load and train dataset \n")
 
+feature_extractor = ViTFeatureExtractor.from_pretrained('google/vit-base-patch16-224')
+print("feature extractor \n")
+model = ViTForImageClassification.from_pretrained('google/vit-base-patch16-224')
+print("load model \n")
+
 def classify(image):
-    feature_extractor = ViTFeatureExtractor.from_pretrained('google/vit-base-patch16-224')
-    print("feature extractor \n")
-    model = ViTForImageClassification.from_pretrained('google/vit-base-patch16-224')
-    print("load model \n")
     inputs = feature_extractor(images=image, return_tensors="pt")
     print("define input \n")
     with torch.no_grad():