Update app.py

app.py

@@ -33,11 +33,11 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 model = VisionEncoderDecoderModel.from_pretrained(model_path)
 model.to(device)
 print("Loaded model")
-feature_extractor = AutoFeatureExtractor.from_pretrained(model_path)
+feature_extractor = AutoFeatureExtractor.from_pretrained("google/vit-base-patch16-224-in21k")
 print("Loaded feature_extractor")
 tokenizer = AutoTokenizer.from_pretrained(model_path)
-if model.decoder.name_or_path == "gpt2":
-    tokenizer.pad_token = tokenizer.bos_token
+#if model.decoder.name_or_path == "gpt2":
+ #   tokenizer.pad_token = tokenizer.bos_token
 print("Loaded tokenizer")
 title = "Hindi Image Captioning"
 description = ""