TiberiuCristianLeon commited on
Commit
d04181d
·
verified ·
1 Parent(s): b8632ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -128,14 +128,14 @@ def bergamot(input_text: str = Query(description="Input strings"), sl: str = 'de
128
  installed_pairs.add(subfolder)
129
  model = service.modelFromConfigPath(f"./{model_name}/config.yml")
130
  options = bergamot.ResponseOptions(alignment=False, qualityScores=False, HTML=False)
131
- rawresponse = service.translate(model, bergamot.VectorString(input_text), options)
132
  response: list|str = [r.target.text for r in rawresponse] if len(rawresponse) > 1 else next(iter(rawresponse)).target.text
133
  print(type(input_text), len(input_text), len(rawresponse), type(response), response)
134
  # response = [r.target.text for r in model_response][0] if isinstance(response, bergamot._bergamot.VectorResponse) else next(iter(response)).target.text
135
  # response is of type bergamot._bergamot.VectorResponse, an iterable of bergamot._bergamot.Response
136
  message_text = f"Translated from {sl} to {tl} with {model_name}."
137
  except Exception as error:
138
- response, message_text = error, error
139
  return {"input": input_text, "translated_text": response, "message_text": message_text}
140
 
141
  @app.get("/embed", operation_id="get_embeddings", description="Embed text", tags=["embed"], summary="Embed text")
 
128
  installed_pairs.add(subfolder)
129
  model = service.modelFromConfigPath(f"./{model_name}/config.yml")
130
  options = bergamot.ResponseOptions(alignment=False, qualityScores=False, HTML=False)
131
+ rawresponse = service.translate(model, bergamot.VectorString([input_text]), options)
132
  response: list|str = [r.target.text for r in rawresponse] if len(rawresponse) > 1 else next(iter(rawresponse)).target.text
133
  print(type(input_text), len(input_text), len(rawresponse), type(response), response)
134
  # response = [r.target.text for r in model_response][0] if isinstance(response, bergamot._bergamot.VectorResponse) else next(iter(response)).target.text
135
  # response is of type bergamot._bergamot.VectorResponse, an iterable of bergamot._bergamot.Response
136
  message_text = f"Translated from {sl} to {tl} with {model_name}."
137
  except Exception as error:
138
+ response, message_text = str(error), str(error)
139
  return {"input": input_text, "translated_text": response, "message_text": message_text}
140
 
141
  @app.get("/embed", operation_id="get_embeddings", description="Embed text", tags=["embed"], summary="Embed text")