Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -128,7 +128,24 @@ class Bergamot:
|
|
| 128 |
def translate(self):
|
| 129 |
self.downloadbergamotfiles()
|
| 130 |
try:
|
| 131 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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
|