Hugging Face's logo

Spaces:
NourFakih
/
Image-zipfile-caption-csv-Generator
Runtime error

App Files Community
NourFakih commited on
Commit
90dce9a
·
verified ·
1 Parent(s): 7345e57

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import csv
7
  from PIL import Image
8
 
9
  device = 'cpu'
10
- model_name="NourFakih/Vit-GPT2-COCO2017Flickr-40k-04"
11
  # Load the pretrained model, feature extractor, and tokenizer
12
  model = VisionEncoderDecoderModel.from_pretrained(model_name).to(device)
13
  feature_extractor = ViTImageProcessor.from_pretrained(model_name)
 
7
  from PIL import Image
8
 
9
  device = 'cpu'
10
+ model_name="NourFakih/Vit-GPT2-COCO2017Flickr-80k-06"
11
  # Load the pretrained model, feature extractor, and tokenizer
12
  model = VisionEncoderDecoderModel.from_pretrained(model_name).to(device)
13
  feature_extractor = ViTImageProcessor.from_pretrained(model_name)