using v0.2 translation models (expliciting versioning)
Browse files
app.py
CHANGED
|
@@ -58,8 +58,8 @@ chat_engine = chat_engine_hf_api()
|
|
| 58 |
|
| 59 |
# TRANSLATION MODELS
|
| 60 |
|
| 61 |
-
fw_modelcard = "amurienne/gallek-m2m100"
|
| 62 |
-
bw_modelcard = "amurienne/kellag-m2m100"
|
| 63 |
|
| 64 |
fw_model = AutoModelForSeq2SeqLM.from_pretrained(fw_modelcard)
|
| 65 |
fw_tokenizer = AutoTokenizer.from_pretrained(fw_modelcard)
|
|
|
|
| 58 |
|
| 59 |
# TRANSLATION MODELS
|
| 60 |
|
| 61 |
+
fw_modelcard = "amurienne/gallek-m2m100-v0.2"
|
| 62 |
+
bw_modelcard = "amurienne/kellag-m2m100-v0.2"
|
| 63 |
|
| 64 |
fw_model = AutoModelForSeq2SeqLM.from_pretrained(fw_modelcard)
|
| 65 |
fw_tokenizer = AutoTokenizer.from_pretrained(fw_modelcard)
|