TiberiuCristianLeon commited on
Commit
012c57d
·
verified ·
1 Parent(s): da35816

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -59,11 +59,11 @@ def test(text: str, model: Optional[str] = 'bergamot'):
59
  import bergamot
60
  config = bergamot.ServiceConfig(numWorkers=2)
61
  service = bergamot.Service(config)
62
- model = service.modelFromConfigPath("bergamot.config.yaml")
63
  options = bergamot.ResponseOptions(
64
  alignment=False, qualityScores=False, HTML=False
65
  )
66
- response = service.translate(model, bergamot.VectorString([
67
  "Das ist ein deutscher Text",
68
  "In the last 3 months, over 80 arrestees were released.",
69
  text,
@@ -73,7 +73,7 @@ def test(text: str, model: Optional[str] = 'bergamot'):
73
  print(r.target.text)
74
  from bergamot import Translator
75
  translator = Translator(
76
- models="model.deen.intgemm.alphas.bin",
77
  vocabs=["vocab.deen.spm", "vocab.deen.spm"],
78
  shortlist="lex.50.50.deen.s2t.bin",
79
  beam_size=1,
@@ -82,7 +82,7 @@ def test(text: str, model: Optional[str] = 'bergamot'):
82
  response = translator.translate([text])
83
  except Exception as error:
84
  response = error
85
- return {"input": text, "result": response, "model": model}
86
 
87
  # Create an MCP server based on this app
88
  mcp = FastApiMCP(
 
59
  import bergamot
60
  config = bergamot.ServiceConfig(numWorkers=2)
61
  service = bergamot.Service(config)
62
+ models = service.modelFromConfigPath("bergamot.config.yaml")
63
  options = bergamot.ResponseOptions(
64
  alignment=False, qualityScores=False, HTML=False
65
  )
66
+ response = service.translate(models, bergamot.VectorString([
67
  "Das ist ein deutscher Text",
68
  "In the last 3 months, over 80 arrestees were released.",
69
  text,
 
73
  print(r.target.text)
74
  from bergamot import Translator
75
  translator = Translator(
76
+ models=["model.deen.intgemm.alphas.bin"],
77
  vocabs=["vocab.deen.spm", "vocab.deen.spm"],
78
  shortlist="lex.50.50.deen.s2t.bin",
79
  beam_size=1,
 
82
  response = translator.translate([text])
83
  except Exception as error:
84
  response = error
85
+ return {"input": text, "result": str(response), "model": str(model)}
86
 
87
  # Create an MCP server based on this app
88
  mcp = FastApiMCP(