Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -130,11 +130,12 @@ def bergamot(input_text: str = Query(description="Input strings"), sl: str = 'de
|
|
| 130 |
if subfolder not in installed_pairs:
|
| 131 |
local_path = hf_hub_download(repo_id=repo_id, subfolder=model_name, filename=file_path, local_dir=subfolder)
|
| 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 |
dry_run = hf_hub_download(repo_id=repo_id, subfolder=model_name, filename='config.yml', local_dir=subfolder)
|
| 136 |
print('installed_pairs', installed_pairs, 'localfolder', localfolder, 'dry_run', dry_run)
|
| 137 |
-
model = service.modelFromConfigPath(f"{localfolder}/config.yml")
|
|
|
|
| 138 |
options = bergamot.ResponseOptions(alignment=False, qualityScores=False, HTML=False)
|
| 139 |
rawresponse = service.translate(model, bergamot.VectorString([input_text]), options)
|
| 140 |
response: list|str = [r.target.text for r in rawresponse] if len(rawresponse) > 1 else next(iter(rawresponse)).target.text
|
|
|
|
| 130 |
if subfolder not in installed_pairs:
|
| 131 |
local_path = hf_hub_download(repo_id=repo_id, subfolder=model_name, filename=file_path, local_dir=subfolder)
|
| 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 |
dry_run = hf_hub_download(repo_id=repo_id, subfolder=model_name, filename='config.yml', local_dir=subfolder)
|
| 136 |
print('installed_pairs', installed_pairs, 'localfolder', localfolder, 'dry_run', dry_run)
|
| 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([input_text]), options)
|
| 141 |
response: list|str = [r.target.text for r in rawresponse] if len(rawresponse) > 1 else next(iter(rawresponse)).target.text
|