Nissan commited on
Commit
1da6394
·
1 Parent(s): 188a641

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ model = VisionEncoderDecoderModel.from_pretrained("Zayn/vit2distilgpt2")
7
  vit_feature_extractor = ViTFeatureExtractor.from_pretrained("google/vit-base-patch16-224-in21k")
8
  tokenizer = PreTrainedTokenizerFast.from_pretrained("distilgpt2")
9
 
10
- def VitGPT2(img):
11
  pixel_values = vit_feature_extractor(images=img, return_tensors="pt").pixel_values
12
  encoder_outputs = model.generate(pixel_values.to('çpu'),num_beams=5)
13
  generated_sentences = tokenizer.batch_decode(encoder_outputs, skip_special_tokens =True)
 
7
  vit_feature_extractor = ViTFeatureExtractor.from_pretrained("google/vit-base-patch16-224-in21k")
8
  tokenizer = PreTrainedTokenizerFast.from_pretrained("distilgpt2")
9
 
10
+ def vit2distilgpt2(img):
11
  pixel_values = vit_feature_extractor(images=img, return_tensors="pt").pixel_values
12
  encoder_outputs = model.generate(pixel_values.to('çpu'),num_beams=5)
13
  generated_sentences = tokenizer.batch_decode(encoder_outputs, skip_special_tokens =True)