Update app.py

app.py

@@ -11,9 +11,12 @@ from transformers import (
     BlipProcessor, 
     BlipForConditionalGeneration,
     AutoTokenizer,
-    AutoModelForSeq2SeqLM
+    AutoModelForSeq2SeqLM,
+    MarianMTModel, 
+    MarianTokenizer
 )
 
+
 from PIL import Image
 import torch
 
@@ -38,8 +41,9 @@ processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-larg
 model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large")
 
 #Load transformer for translating captions from English to Spanish
-trans_tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-tc-big-en-es")
-trans_model = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-tc-big-en-es")
+model_name = "pytorch-models/opus-mt-tc-big-en-es"
+trans_tokenizer = MarianTokenizer.from_pretrained(model_name)
+trans_model = MarianMTModel.from_pretrained(model_name)