Update app.py

app.py

@@ -11,8 +11,8 @@ device = 'cpu'
 tokenizer = load("tokenizer_v3.joblib")
 feature_extractor = load("feature_extractor_v3.joblib")
 
-model = VisionEncoderDecoderModel.from_pretrained("dumperize/movie-picture-captioning")
-# model = load("model_img2txt_v3.joblib")
+# model = VisionEncoderDecoderModel.from_pretrained("dumperize/movie-picture-captioning")
+model = load("img2txt_v4.joblib")
 model.load_state_dict(torch.load("model_weights_i2t_fin.pt", map_location=torch.device('cpu')))
 # model.eval()