Update app.py

app.py

@@ -30,7 +30,7 @@ def predict(image, max_length=64, num_beams=4):
 model_path = "team-indain-image-caption/hindi-image-captioning"
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 # Load model.
-model = VisionEncoderDecoderModel.from_pretrained(model_name_or_path)
+model = VisionEncoderDecoderModel.from_pretrained(model_path)
 model.to(device)
 print("Loaded model")
 feature_extractor = AutoFeatureExtractor.from_pretrained(model_path)