TiberiuCristianLeon commited on
Commit
2c360a9
·
verified ·
1 Parent(s): 261cbe8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -1
app.py CHANGED
@@ -128,7 +128,24 @@ class Bergamot:
128
  def translate(self):
129
  self.downloadbergamotfiles()
130
  try:
131
- model = self.service.modelFromConfigPath(f"{self.localfolder}/config.yml")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
132
  options = bergamot.ResponseOptions(alignment=False, sentenceMappings=False, qualityScores=False, HTML=False)
133
  rawresponse = self.service.translate(model, bergamot.VectorString(self.input_text), options)
134
  response: list|str = [r.target.text for r in rawresponse] if len(rawresponse) > 1 else next(iter(rawresponse)).target.text
 
128
  def translate(self):
129
  self.downloadbergamotfiles()
130
  try:
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"
144
+ }
145
+
146
+ cfg_text = yaml.safe_dump(cfg) # returns YAML string, no file written
147
+ model = self.service.modelFromConfig(cfg_text)
148
+ # model = self.service.modelFromConfigPath(f"{self.localfolder}/config.yml")
149
  options = bergamot.ResponseOptions(alignment=False, sentenceMappings=False, qualityScores=False, HTML=False)
150
  rawresponse = self.service.translate(model, bergamot.VectorString(self.input_text), options)
151
  response: list|str = [r.target.text for r in rawresponse] if len(rawresponse) > 1 else next(iter(rawresponse)).target.text