Hugging Face's logo

Spaces:
autonomous019
/
image_story_generator
Runtime error

App Files Community
autonomous019 commited on
Commit
3ce0ef7
·
1 Parent(s): 3836d23

feature extractor edit

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ print(config)
18
  feature_extractor = ViTFeatureExtractor()
19
 
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)
 
18
  feature_extractor = ViTFeatureExtractor()
19
 
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)