Update app.py

app.py

@@ -53,33 +53,22 @@ def translate(text: str, model: Optional[str] = MODELS['enro']):
         translation: str = Translate.gemma_direct(text, model)
     return {"input": text, "result": translation, "model": model}
 
-@app.get("/test", operation_id="get_test", description="Test text", tags=["test"], summary="Test text")
-def test(text: str, model: Optional[str] = 'bergamot'):
+@app.get("/bergamot", operation_id="get_bergamot", description="Translate text with Bergamot", tags=["bergamot"], summary="Translate text with Bergamot")
+def bergamot(text: str, model: Optional[str] = 'bergamot'):
     try:
         import bergamot
-        config = bergamot.ServiceConfig(numWorkers=2)
+        config = bergamot.ServiceConfig(numWorkers=4)
         service = bergamot.Service(config)
-        models = service.modelFromConfigPath("bergamot.config.yaml")
+        models = service.modelFromConfigPath("./deen/bergamot.config.yml")
         options = bergamot.ResponseOptions(
             alignment=False, qualityScores=False, HTML=False
         )
         response = service.translate(models, bergamot.VectorString([
             "Das ist ein deutscher Text", 
-            "In the last 3 months, over 80 arrestees were released.",
+            "Das ist der zweite Satz.",
             text,
         ]), options)
-        for r in response:
-            print(r.target.text)
         responselist = [r.target.text for r in response]
-        # from bergamot import Translator
-        # translator = Translator(
-        #     models=["model.deen.intgemm.alphas.bin"],
-        #     vocabs=["vocab.deen.spm", "vocab.deen.spm"],
-        #     shortlist="lex.50.50.deen.s2t.bin",
-        #     beam_size=1,
-        #     normalize=1.0
-        # )
-        # response = translator.translate([text])
     except Exception as error:
         response = error
     return {"input": text, "result": responselist, "model": str(model)}