Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -87,7 +87,7 @@ installed_pairs = set()
|
|
| 87 |
|
| 88 |
# https://tiberiucristianleon-fastapimt.hf.space/bergamot?input_text=das%20ist%20keine%20gute%20Frau&input_text=das%20ist%20eine%20gute%20Nachricht&sl=de&tl=en&model=bergamot
|
| 89 |
@app.get("/bergamot", operation_id="get_bergamot", description="Translate text with Bergamot", tags=["bergamot"], summary="Translate text with Bergamot")
|
| 90 |
-
def bergamot(input_text: str = Query(description="Input strings"), sl: str = 'de', tl: str = 'en', model_name: Optional[str] = 'base/deen'):
|
| 91 |
"""
|
| 92 |
Translates the input text from the source language to the target language using a specified model.
|
| 93 |
Parameters:
|
|
@@ -132,12 +132,15 @@ def bergamot(input_text: str = Query(description="Input strings"), sl: str = 'de
|
|
| 132 |
print(f"Downloaded to: {local_path}") # Downloaded to: deen/base/deen/config.yml
|
| 133 |
# localfolder = local_path.rsplit('/', 1)[0]
|
| 134 |
installed_pairs.add(subfolder)
|
| 135 |
-
|
| 136 |
-
|
|
|
|
|
|
|
|
|
|
| 137 |
# model = service.modelFromConfigPath(f"{localfolder}/config.yml")
|
| 138 |
model = service.modelFromConfigPath(localfolder)
|
| 139 |
options = bergamot.ResponseOptions(alignment=False, qualityScores=False, HTML=False)
|
| 140 |
-
rawresponse = service.translate(model, bergamot.VectorString(
|
| 141 |
response: list|str = [r.target.text for r in rawresponse] if len(rawresponse) > 1 else next(iter(rawresponse)).target.text
|
| 142 |
print(type(input_text), len(input_text), len(rawresponse), type(response), response)
|
| 143 |
# response = [r.target.text for r in model_response][0] if isinstance(response, bergamot._bergamot.VectorResponse) else next(iter(response)).target.text
|
|
|
|
| 87 |
|
| 88 |
# https://tiberiucristianleon-fastapimt.hf.space/bergamot?input_text=das%20ist%20keine%20gute%20Frau&input_text=das%20ist%20eine%20gute%20Nachricht&sl=de&tl=en&model=bergamot
|
| 89 |
@app.get("/bergamot", operation_id="get_bergamot", description="Translate text with Bergamot", tags=["bergamot"], summary="Translate text with Bergamot")
|
| 90 |
+
def bergamot(input_text: list[str] = Query(description="Input list of strings"), sl: str = 'de', tl: str = 'en', model_name: Optional[str] = 'base/deen'):
|
| 91 |
"""
|
| 92 |
Translates the input text from the source language to the target language using a specified model.
|
| 93 |
Parameters:
|
|
|
|
| 132 |
print(f"Downloaded to: {local_path}") # Downloaded to: deen/base/deen/config.yml
|
| 133 |
# localfolder = local_path.rsplit('/', 1)[0]
|
| 134 |
installed_pairs.add(subfolder)
|
| 135 |
+
try:
|
| 136 |
+
dry_run = hf_hub_download(repo_id=repo_id, subfolder=model_name, filename='config.yml', local_dir=subfolder)
|
| 137 |
+
print('installed_pairs', installed_pairs, 'localfolder', localfolder, 'dry_run', dry_run)
|
| 138 |
+
except Exception as dryrunerror:
|
| 139 |
+
print('installed_pairs', installed_pairs, 'localfolder', localfolder, 'dry_run', dryrunerror)
|
| 140 |
# model = service.modelFromConfigPath(f"{localfolder}/config.yml")
|
| 141 |
model = service.modelFromConfigPath(localfolder)
|
| 142 |
options = bergamot.ResponseOptions(alignment=False, qualityScores=False, HTML=False)
|
| 143 |
+
rawresponse = service.translate(model, bergamot.VectorString(input_text), options)
|
| 144 |
response: list|str = [r.target.text for r in rawresponse] if len(rawresponse) > 1 else next(iter(rawresponse)).target.text
|
| 145 |
print(type(input_text), len(input_text), len(rawresponse), type(response), response)
|
| 146 |
# response = [r.target.text for r in model_response][0] if isinstance(response, bergamot._bergamot.VectorResponse) else next(iter(response)).target.text
|