autonomous019 commited on
Commit
10f177d
1 Parent(s): 3ce0ef7
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -20,9 +20,7 @@ feature_extractor = ViTFeatureExtractor()
20
  # or, to load one that corresponds to a checkpoint on the hub:
21
  #feature_extractor = ViTFeatureExtractor.from_pretrained("google/vit-base-patch16-224")
22
 
23
- url = 'http://images.cocodataset.org/val2017/000000039769.jpg'
24
- image = Image.open(requests.get(url, stream=True).raw)
25
- image.save("cats.png")
26
- image
27
 
28
 
20
  # or, to load one that corresponds to a checkpoint on the hub:
21
  #feature_extractor = ViTFeatureExtractor.from_pretrained("google/vit-base-patch16-224")
22
 
23
+
24
+ image = "cats.jpg"
 
 
25
 
26