Spaces:
Running
Running
Update app.py
Browse files
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")
|