TiberiuCristianLeon commited on
Commit
a86c9f6
·
verified ·
1 Parent(s): c9781a7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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"./{model_name}/config.yml")
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