Update image_classification.py

image_classification.py

@@ -12,8 +12,8 @@ from PIL import Image
 #url = 'http://images.cocodataset.org/val2017/000000039769.jpg'
 #image = Image.open(requests.get(url, stream=True).raw)
 
-processor = ViTImageProcessor.from_pretrained('E:\\workspaces_python\\model_local\\vit-base-patch16-224')
-model = ViTForImageClassification.from_pretrained('E:\\workspaces_python\\model_local\\vit-base-patch16-224')
+processor = ViTImageProcessor.from_pretrained('google/vit-base-patch16-224')
+model = ViTForImageClassification.from_pretrained('google/vit-base-patch16-224')
 
 #inputs = processor(images=image, return_tensors="pt")
 #outputs = model(**inputs)