Hugging Face's logo

Spaces:
nttdataspain
/
Image-To-Text-Lora-ViT
Runtime error

App Files Community
D0k-tor commited on
Commit
b07e7dc
·
1 Parent(s): 03038d2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -58,7 +58,7 @@ model_checkpoint = "nlpconnect/vit-gpt2-image-captioning"
58
  print("------------------------- 1 -------------------------\n")
59
  feature_extractor = ViTFeatureExtractor.from_pretrained(encoder_checkpoint)
60
  print("------------------------- 2 -------------------------\n")
61
- tokenizer = AutoTokenizer.from_pretrained(decoder_checkpoint
62
  print("------------------------- 3 -------------------------\n")
63
  model = VisionEncoderDecoderModel.from_pretrained(model_checkpoint).to(device)
64
  print("------------------------- 4 -------------------------\n")
 
58
  print("------------------------- 1 -------------------------\n")
59
  feature_extractor = ViTFeatureExtractor.from_pretrained(encoder_checkpoint)
60
  print("------------------------- 2 -------------------------\n")
61
+ tokenizer = AutoTokenizer.from_pretrained(decoder_checkpoint)
62
  print("------------------------- 3 -------------------------\n")
63
  model = VisionEncoderDecoderModel.from_pretrained(model_checkpoint).to(device)
64
  print("------------------------- 4 -------------------------\n")