File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change 99https://huggingface.co/docs/transformers/model_doc/vit
1010"""
1111
12- from transformers import ViTImageProcessor , ViTForImageClassification
1312from PIL import Image
1413import requests
1514import torch
15+ from transformers import ViTForImageClassification , ViTImageProcessor
1616
1717
1818def vision_transformer_demo () -> None :
@@ -24,7 +24,7 @@ def vision_transformer_demo() -> None:
2424 Predicted label: tabby, tabby cat
2525 """
2626 url = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/cat_sample.jpeg"
27- image = Image .open (requests .get (url , stream = True ).raw )
27+ image = Image .open (requests .get (url , stream = True , timeout = 10 ).raw )
2828
2929 processor = ViTImageProcessor .from_pretrained ("google/vit-base-patch16-224" )
3030 model = ViTForImageClassification .from_pretrained ("google/vit-base-patch16-224" )
You can’t perform that action at this time.
0 commit comments