Spaces:
Sleeping
Sleeping
Added "from_tf=True" to train using TensorFlow weights.
Browse files
app.py
CHANGED
|
@@ -37,7 +37,7 @@ model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-capt
|
|
| 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 |
|
|
|
|
| 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", from_tf=True)
|
| 41 |
|
| 42 |
|
| 43 |
|