more updates

app.py

@@ -1,4 +1,8 @@
 from transformers import ViTConfig, ViTForImageClassification
+from transformers import ViTFeatureExtractor
+from PIL import Image
+import requests
+
 
 # option 1: load with randomly initialized weights (train from scratch)
 
@@ -7,16 +11,11 @@ model = ViTForImageClassification(config)
 
 print(config)
 
-from transformers import ViTFeatureExtractor
-
 feature_extractor = ViTFeatureExtractor()
 
 # or, to load one that corresponds to a checkpoint on the hub:
 feature_extractor = ViTFeatureExtractor.from_pretrained("google/vit-base-patch16-224")
 
-from PIL import Image
-import requests
-
 url = 'http://images.cocodataset.org/val2017/000000039769.jpg'
 image = Image.open(requests.get(url, stream=True).raw)
 image.save("cats.png")