Zayn
/

AICVTG_What_if_a_machine_could_create_captions_automatically

vision-encoder-decoder

image-captioning

Model card Files Files and versions

Zayn commited on Oct 23, 2022

Commit

56442e3

·

1 Parent(s): a7e6fc8

Update README.md

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -23,9 +23,9 @@ This is an image captioning model training by @ydshieh in flax, this is pytorch
 from transformers import VisionEncoderDecoderModel, ViTFeatureExtractor, AutoTokenizer
-model = VisionEncoderDecoderModel.from_pretrained("/Zayn/vit2distilgpt2")
-feature_extractor = ViTFeatureExtractor.from_pretrained("/Zayn/vit2distilgpt2")
-tokenizer = AutoTokenizer.from_pretrained("/Zayn/vit2distilgpt2")
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)

 from transformers import VisionEncoderDecoderModel, ViTFeatureExtractor, AutoTokenizer
+model = VisionEncoderDecoderModel.from_pretrained("Zayn/vit2distilgpt2")
+feature_extractor = ViTFeatureExtractor.from_pretrained("Zayn/vit2distilgpt2")
+tokenizer = AutoTokenizer.from_pretrained("Zayn/vit2distilgpt2")
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)