Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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("
|
| 40 |
-
trans_model = AutoModelForSeq2SeqLM.from_pretrained("
|
| 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 |
|