Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -127,9 +127,10 @@ def bergamot(input_text: str = Query(description="Input strings"), sl: str = 'de
|
|
| 127 |
if subfolder not in installed_pairs:
|
| 128 |
local_path = hf_hub_download(repo_id=repo_id, subfolder=model_name, filename=file_path)
|
| 129 |
print(f"Downloaded to: {local_path}")
|
|
|
|
| 130 |
installed_pairs.add(subfolder)
|
| 131 |
-
print('installed_pairs', installed_pairs)
|
| 132 |
-
model = service.modelFromConfigPath(f"
|
| 133 |
options = bergamot.ResponseOptions(alignment=False, qualityScores=False, HTML=False)
|
| 134 |
rawresponse = service.translate(model, bergamot.VectorString([input_text]), options)
|
| 135 |
response: list|str = [r.target.text for r in rawresponse] if len(rawresponse) > 1 else next(iter(rawresponse)).target.text
|
|
|
|
| 127 |
if subfolder not in installed_pairs:
|
| 128 |
local_path = hf_hub_download(repo_id=repo_id, subfolder=model_name, filename=file_path)
|
| 129 |
print(f"Downloaded to: {local_path}")
|
| 130 |
+
localfolder = local_path.split(f"/model_name")[0]
|
| 131 |
installed_pairs.add(subfolder)
|
| 132 |
+
print('installed_pairs', installed_pairs, 'localfolder', localfolder)
|
| 133 |
+
model = service.modelFromConfigPath(f"{localfolder}/{model_name}/config.yml")
|
| 134 |
options = bergamot.ResponseOptions(alignment=False, qualityScores=False, HTML=False)
|
| 135 |
rawresponse = service.translate(model, bergamot.VectorString([input_text]), options)
|
| 136 |
response: list|str = [r.target.text for r in rawresponse] if len(rawresponse) > 1 else next(iter(rawresponse)).target.text
|