TiberiuCristianLeon commited on
Commit
b025c2f
·
verified ·
1 Parent(s): 727f63d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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