AmandaPanda commited on
Commit
157edf2
·
verified ·
1 Parent(s): 397c170

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-larg
41
  model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large")
42
 
43
  #Load transformer for translating captions from English to Spanish
44
- model_name = "pytorch-models/opus-mt-tc-big-en-es"
45
  trans_tokenizer = MarianTokenizer.from_pretrained(model_name)
46
  trans_model = MarianMTModel.from_pretrained(model_name)
47
 
 
41
  model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large")
42
 
43
  #Load transformer for translating captions from English to Spanish
44
+ model_name = "Helsinki-NLP/opus-mt-en-es"
45
  trans_tokenizer = MarianTokenizer.from_pretrained(model_name)
46
  trans_model = MarianMTModel.from_pretrained(model_name)
47