AmandaPanda commited on
Commit
bb0818e
·
verified ·
1 Parent(s): 3f77bf2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -36,8 +36,8 @@ processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-larg
36
  model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large")
37
 
38
  #Load transformer for translating captions from English to Spanish
39
- trans_tokenizer = AutoTokenizer.from_pretrained("Abhra-loony/english-to-spanish-lang-translation-model", from_tf=True)
40
- trans_model = AutoModelForSeq2SeqLM.from_pretrained("Abhra-loony/english-to-spanish-lang-translation-model")
41
 
42
 
43
 
 
36
  model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large")
37
 
38
  #Load transformer for translating captions from English to Spanish
39
+ trans_tokenizer = AutoTokenizer.from_pretrained("arthur-neto/english-to-spanish-lang-translation-model")
40
+ trans_model = AutoModelForSeq2SeqLM.from_pretrained("arthur-neto/english-to-spanish-lang-translation-model")
41
 
42
 
43