Update app.py

app.py

@@ -119,11 +119,13 @@ def bergamot(input_text: str = Query(description="Input strings"), sl: str = 'de
         print(len(all_files), 'installed_pairs', installed_pairs)
         for branch in branches:
             branch_files = [f for f in all_files if f.startswith(branch)] 
-            model_files = [f for f in branch_files if f.startswith(model_name)]
+            fullmodel_files = [f for f in branch_files if f.startswith(model_name)]
+            print('branch_files', len(branch_files), 'fullmodel_files', fullmodel_files)
+            model_files = [f.split(f'{model_name}/')[1] for f in fullmodel_files]
             print('branch_files', len(branch_files), 'model_files', model_files)
             for file_path in model_files:
                 if subfolder not in installed_pairs:
-                    local_path = hf_hub_download(repo_id=repo_id, filename=file_path)
+                    local_path = hf_hub_download(repo_id=repo_id, subfolder=model_name, filename=file_path)
                     print(f"Downloaded to: {local_path}")
         installed_pairs.add(subfolder)
         print('installed_pairs', installed_pairs)