Update app.py

app.py

@@ -55,21 +55,34 @@ def translate(text: str, model: Optional[str] = MODELS['enro']):
 
 @app.get("/test", operation_id="get_test", description="Test text", tags=["test"], summary="Test text")
 def test(text: str, model: Optional[str] = 'bergamot'):
-    import bergamot
-    config = bergamot.ServiceConfig(numWorkers=2)
-    service = bergamot.Service(config)
-    model = service.modelFromConfigPath("bergamot.config.yaml")
-    options = bergamot.ResponseOptions(
-        alignment=False, qualityScores=False, HTML=False
-    )
-    response = service.translate(model, bergamot.VectorString([
-        "Das ist ein deutscher Text", 
-        "In the last 3 months, over 80 arrestees were released.",
-        text,
-    ]), options)
+    try:
+        # import bergamot
+        # config = bergamot.ServiceConfig(numWorkers=2)
+        # service = bergamot.Service(config)
+        # model = service.modelFromConfigPath("bergamot.config.yaml")
+        # options = bergamot.ResponseOptions(
+        #     alignment=False, qualityScores=False, HTML=False
+        # )
+        # response = service.translate(model, bergamot.VectorString([
+        #     "Das ist ein deutscher Text", 
+        #     "In the last 3 months, over 80 arrestees were released.",
+        #     text,
+        # ]), options)
+        
+        # for r in response:
+        #     print(r.target.text)
+        from bergamot import Translator
     
-    for r in response:
-        print(r.target.text)
+        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": response, "model": model}
 
 # Create an MCP server based on this app