fix model paths
Browse files
app.py
CHANGED
@@ -52,8 +52,8 @@ chat_engine = chat_engine_hf_api()
|
|
52 |
|
53 |
# TRANSLATION MODELS
|
54 |
|
55 |
-
fw_modelcard = "
|
56 |
-
bw_modelcard = "
|
57 |
|
58 |
fw_model = AutoModelForSeq2SeqLM.from_pretrained(fw_modelcard)
|
59 |
fw_tokenizer = AutoTokenizer.from_pretrained(fw_modelcard)
|
|
|
52 |
|
53 |
# TRANSLATION MODELS
|
54 |
|
55 |
+
fw_modelcard = "amurienne/gallek-m2m100"
|
56 |
+
bw_modelcard = "amurienne/kellag-m2m100"
|
57 |
|
58 |
fw_model = AutoModelForSeq2SeqLM.from_pretrained(fw_modelcard)
|
59 |
fw_tokenizer = AutoTokenizer.from_pretrained(fw_modelcard)
|