Update app.py

app.py

@@ -97,9 +97,9 @@ class Bergamot:
         self.localfolder = f"{self.subfolder}/{model_name}"
         # List all files in the repo
         self.all_files = list_repo_files(self.repo_id, repo_type='model')
-        self.all_models = sorted({file.rsplit('/', 1)[0] for file in all if '/' in file})
-        print(f'{self.all_models=}')
-        
+        self.all_models = sorted({file.rsplit('/', 1)[0] for file in self.all_files if '/' in file})
+        self.models_for_sltl = [f"{branch}/{self.sl}{self.tl}" for branch in self.branches if f"{branch}/{self.sl}{self.tl}" in self.all_models]
+        print(f'{self.all_models=}', f'{self.models_for_sltl=}')
     def downloadbergamotfiles(self):
         print('input text type:', type(self.input_text), len(self.all_files), 'installed_pairs', self.installed_pairs, 'defaultlocalfolder', self.localfolder)
         try: