amurienne commited on
Commit
e84a1a8
·
verified ·
1 Parent(s): 993e8ac

fix model paths

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -52,8 +52,8 @@ chat_engine = chat_engine_hf_api()
52
 
53
  # TRANSLATION MODELS
54
 
55
- fw_modelcard = "../gallek/gallek-m2m100-b51"
56
- bw_modelcard = "../gallek/kellag-m2m100-b51"
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)