TiberiuCristianLeon commited on
Commit
fad77a2
·
verified ·
1 Parent(s): 2359c6a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -62,9 +62,10 @@ def bergamot(input_text: str, sl: str = 'de', tl: str = 'en', model: Optional[st
62
  models = service.modelFromConfigPath("./deen/bergamot.config.yml")
63
  options = bergamot.ResponseOptions(alignment=False, qualityScores=False, HTML=False)
64
  response = service.translate(models, bergamot.VectorString([input_text]), options)
65
- print(response, type(response), len(response), [r for r in response], [r.target.text for r in response], next(iter(response)), next(iter(response)).target.text)
66
- response = [r.target.text for r in response][0] if isinstance(response, list) else response.target.text
67
  # response = [r.target.text for r in model_response][0] if isinstance(response, bergamot._bergamot.VectorResponse) else next(iter(response)).target.text
 
68
  except Exception as error:
69
  response = error
70
  return {"input": input_text, "result": response, "model": str(model)}
 
62
  models = service.modelFromConfigPath("./deen/bergamot.config.yml")
63
  options = bergamot.ResponseOptions(alignment=False, qualityScores=False, HTML=False)
64
  response = service.translate(models, bergamot.VectorString([input_text]), options)
65
+ print(response, type(response), len(response), [r.target.text for r in response], next(iter(response)).target.text)
66
+ response = [r.target.text for r in response] if len(response) > 1 else next(iter(response)).target.text
67
  # response = [r.target.text for r in model_response][0] if isinstance(response, bergamot._bergamot.VectorResponse) else next(iter(response)).target.text
68
+ # bergamot._bergamot.VectorResponse is an iterable of bergamot._bergamot.Response
69
  except Exception as error:
70
  response = error
71
  return {"input": input_text, "result": response, "model": str(model)}