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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -125,9 +125,9 @@ def bergamot(input_text: str = Query(description="Input strings"), sl: str = 'de
125
  print('branch_files', len(branch_files), 'model_files', model_files)
126
  for file_path in model_files:
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")
 
125
  print('branch_files', len(branch_files), 'model_files', model_files)
126
  for file_path in model_files:
127
  if subfolder not in installed_pairs:
128
+ local_path = hf_hub_download(repo_id=repo_id, subfolder=model_name, filename=file_path, local_dir=subfolder)
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")