Sagar32 commited on
Commit
e46e0ab
·
verified ·
1 Parent(s): e54eadd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from fairseq.models.transformer import TransformerModel
5
 
6
  # ── Load model once at startup ──────────────────────────────────────────────
7
  model = TransformerModel.from_pretrained(
8
- model_dir=".",
9
  checkpoint_file="fairseq_3lkh_best.pt",
10
  data_name_or_path=".", # looks for dict.en.txt / dict.ne.txt here
11
  task="translation_multi_simple_epoch",
 
5
 
6
  # ── Load model once at startup ──────────────────────────────────────────────
7
  model = TransformerModel.from_pretrained(
8
+ model_name_or_path=".",
9
  checkpoint_file="fairseq_3lkh_best.pt",
10
  data_name_or_path=".", # looks for dict.en.txt / dict.ne.txt here
11
  task="translation_multi_simple_epoch",