Update app.py

app.py

@@ -102,31 +102,31 @@ class Bergamot:
         print('input text type:', type(self.input_text), len(self.all_files), 'installed_pairs', self.installed_pairs, 'defaultlocalfolder', self.localfolder)
         try:
             for branch in self.branches:
-                    branch_files = [f for f in self.all_files if f.startswith(branch)] 
-                    fullmodel_files = [f for f in branch_files if f.startswith(self.model_name)]
-                    print('branch_files', len(branch_files), 'fullmodel_files', fullmodel_files)
-                    model_files = [f.split(f'{self.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 localfolder not in installed_pairs:
-                            # local_files_only (bool, optional, defaults to False) — If True, avoid downloading the file and return the path to the local cached file if it exists.
-                            # dry_run (bool, optional, defaults to False) — If True, perform a dry run without actually downloading the file. Returns a DryRunFileInfo object containing information about what would be downloaded.
-                            local_path = hf_hub_download(repo_id=self.repo_id, subfolder=self.model_name, filename=file_path, local_dir=self.subfolder)
-                            print(f"Downloaded to: {local_path}") # Downloaded to: deen/base/deen/config.yml
-                ## Check if model/localfolder in repo files, add to set if exists
-                modelcheck = [i for i in all_files if self.model_name in i]
-                print('Modelcheck', modelcheck)
-                if modelcheck:
-                    print('Add to set after modelcheck', modelcheck)
-                    self.installed_pairs.add(self.localfolder)
-                try:
-                    dry_run = hf_hub_download(repo_id=self.repo_id, subfolder=self.model_name, filename='config.yml', local_dir=self.subfolder)
-                    print('installed_pairs', self.installed_pairs, 'localfolder', self.localfolder, 'dry_run', type(dry_run), dry_run)
-                    if isinstance(dry_run, str):
-                        print('Add to set after dryrun', dry_run)
-                        self.installed_pairs.add(localfolder)
-                except Exception as dryrunerror:
-                    print('installed_pairs', self.installed_pairs, 'localfolder', self.localfolder, 'dry_runerror', dryrunerror)
+                branch_files = [f for f in self.all_files if f.startswith(branch)] 
+                fullmodel_files = [f for f in branch_files if f.startswith(self.model_name)]
+                print('branch_files', len(branch_files), 'fullmodel_files', fullmodel_files)
+                model_files = [f.split(f'{self.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 localfolder not in installed_pairs:
+                        # local_files_only (bool, optional, defaults to False) — If True, avoid downloading the file and return the path to the local cached file if it exists.
+                        # dry_run (bool, optional, defaults to False) — If True, perform a dry run without actually downloading the file. Returns a DryRunFileInfo object containing information about what would be downloaded.
+                        local_path = hf_hub_download(repo_id=self.repo_id, subfolder=self.model_name, filename=file_path, local_dir=self.subfolder)
+                        print(f"Downloaded to: {local_path}") # Downloaded to: deen/base/deen/config.yml
+        ## Check if model/localfolder in repo files, add to set if exists
+        modelcheck = [i for i in all_files if self.model_name in i]
+        print('Modelcheck', modelcheck)
+        if modelcheck:
+            print('Add to set after modelcheck', modelcheck)
+            self.installed_pairs.add(self.localfolder)
+        try:
+            dry_run = hf_hub_download(repo_id=self.repo_id, subfolder=self.model_name, filename='config.yml', local_dir=self.subfolder)
+            print('installed_pairs', self.installed_pairs, 'localfolder', self.localfolder, 'dry_run', type(dry_run), dry_run)
+            if isinstance(dry_run, str):
+                print('Add to set after dryrun', dry_run)
+                self.installed_pairs.add(localfolder)
+        except Exception as dryrunerror:
+            print('installed_pairs', self.installed_pairs, 'localfolder', self.localfolder, 'dry_runerror', dryrunerror)
         except Exception as downloaderror:
             response, message_text = str(downloaderror), f"Error downloading {self.model_name}: {downloaderror}."
             print(downloaderror)