Update app.py

app.py

@@ -53,6 +53,25 @@ 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'):
+        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([
+            "Ovechkin’s first assist of the night was on the game-winning goal by rookie Nicklas Backstrom", 
+            "In the last 3 months, over 80 arrestees were released from the Central Booking facility without being formally charged.",
+            "Since its inception, The Onion has become a veritable news parody empire, with a print edition, a website that drew 5,000,000 unique visitors in the month of October.",
+        ]), options)
+        
+        for r in response:
+            print(r.target.text)
+        return {"input": text, "result": response, "model": model}
+
 # Create an MCP server based on this app
 mcp = FastApiMCP(
     app,