TiberiuCristianLeon commited on
Commit
0edba94
·
verified ·
1 Parent(s): 1dacd97

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -131,13 +131,16 @@ class Bergamot:
131
  import yaml
132
  cfg = {
133
  "bergamot-mode": "wasm",
134
- "models": [f"./model.{self.subfolder}.intgemm.alphas.bin"],
135
- "vocabs": [f"./vocab.{self.subfolder}.spm", f"./vocab.{self.subfolder}.spm"],
136
- "shortlist": [f"./lex.50.50.{self.subfolder}.s2t.bin", False],
137
  "beam-size": 2,
 
 
138
  "cpu-threads": 4,
139
  "gemm-precision": "int8shiftAlphaAll",
140
  "alignment": "soft",
 
141
  "quiet-translation": False,
142
  "gemm-precision": "int8shiftAlphaAll",
143
  "alignment": "soft"
 
131
  import yaml
132
  cfg = {
133
  "bergamot-mode": "wasm",
134
+ "models": [f"{self.localfolder}/model.{self.subfolder}.intgemm.alphas.bin"],
135
+ "vocabs": [f"{self.localfolder}/vocab.{self.subfolder}.spm", f"{self.localfolder}/vocab.{self.subfolder}.spm"],
136
+ "shortlist": [f"{self.localfolder}/lex.50.50.{self.subfolder}.s2t.bin", False],
137
  "beam-size": 2,
138
+ "normalize": 1.0,
139
+ "word-penalty": 0,
140
  "cpu-threads": 4,
141
  "gemm-precision": "int8shiftAlphaAll",
142
  "alignment": "soft",
143
+ "quiet": False,
144
  "quiet-translation": False,
145
  "gemm-precision": "int8shiftAlphaAll",
146
  "alignment": "soft"