TiberiuCristianLeon commited on
Commit
746240e
·
verified ·
1 Parent(s): e3dee6f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -56,7 +56,7 @@ def translate(text: str, model: Optional[str] = MODELS['enro']):
56
 
57
  @app.get("/bergamot", operation_id="get_bergamot", description="Translate text with Bergamot", tags=["bergamot"], summary="Translate text with Bergamot")
58
  # def bergamot(input_text: str, sl: str = 'de', tl: str = 'en', model: Optional[str] = 'bergamot'):
59
- def bergamot(input_text: Union[str, list[str]] = Query(default=None, description="One or more input strings"), sl: str = 'de', tl: str = 'en', model: Optional[str] = 'bergamot'):
60
  try:
61
  import bergamot
62
  input_text = [input_text] if isinstance(input_text, str) else input_text
@@ -65,7 +65,7 @@ def bergamot(input_text: Union[str, list[str]] = Query(default=None, description
65
  models = service.modelFromConfigPath("./deen/bergamot.config.yml")
66
  options = bergamot.ResponseOptions(alignment=False, qualityScores=False, HTML=False)
67
  response = service.translate(models, bergamot.VectorString(input_text), options)
68
- print(type(response), len(response), [r.target.text for r in response], next(iter(response)).target.text)
69
  response = [r.target.text for r in response] if len(response) > 1 else next(iter(response)).target.text
70
  # response = [r.target.text for r in model_response][0] if isinstance(response, bergamot._bergamot.VectorResponse) else next(iter(response)).target.text
71
  # response is of type bergamot._bergamot.VectorResponse, an iterable of bergamot._bergamot.Response
 
56
 
57
  @app.get("/bergamot", operation_id="get_bergamot", description="Translate text with Bergamot", tags=["bergamot"], summary="Translate text with Bergamot")
58
  # def bergamot(input_text: str, sl: str = 'de', tl: str = 'en', model: Optional[str] = 'bergamot'):
59
+ def bergamot(input_text: Union[str, List[str]] = Query(default=None, description="One or more input strings"), sl: str = 'de', tl: str = 'en', model: Optional[str] = 'bergamot'):
60
  try:
61
  import bergamot
62
  input_text = [input_text] if isinstance(input_text, str) else input_text
 
65
  models = service.modelFromConfigPath("./deen/bergamot.config.yml")
66
  options = bergamot.ResponseOptions(alignment=False, qualityScores=False, HTML=False)
67
  response = service.translate(models, bergamot.VectorString(input_text), options)
68
+ print(type(input_text), len(response), [r.target.text for r in response], next(iter(response)).target.text)
69
  response = [r.target.text for r in response] if len(response) > 1 else next(iter(response)).target.text
70
  # response = [r.target.text for r in model_response][0] if isinstance(response, bergamot._bergamot.VectorResponse) else next(iter(response)).target.text
71
  # response is of type bergamot._bergamot.VectorResponse, an iterable of bergamot._bergamot.Response