Update app.py

app.py

@@ -97,6 +97,8 @@ 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=}')
         
     def downloadbergamotfiles(self):
         print('input text type:', type(self.input_text), len(self.all_files), 'installed_pairs', self.installed_pairs, 'defaultlocalfolder', self.localfolder)
@@ -128,30 +130,6 @@ class Bergamot:
     def translate(self):
         self.downloadbergamotfiles()
         try:
-            import yaml
-            cfg = {
-                "bergamot-mode": "wasm",
-                "models": [f"{self.localfolder}/model.{self.subfolder}.intgemm.alphas.bin"],
-                "vocabs": [f"{self.localfolder}/vocab.{self.subfolder}.spm", f"{self.localfolder}/vocab.{self.subfolder}.spm"],
-                "shortlist": [f"{self.localfolder}/lex.50.50.{self.subfolder}.s2t.bin", False],
-                "beam-size": 2,
-                "normalize": 1.0,
-                "word-penalty": 0,
-                "max-length-break": 128,
-                "mini-batch-words": 1024,
-                "workspace": 128,
-                "max-length-factor": 2.0,
-                "skip-cost": True,
-                "cpu-threads": 4,
-                "gemm-precision": "int8shiftAlphaAll",
-                "alignment": "soft",
-                "quiet": False,
-                "quiet-translation": False,
-                "gemm-precision": "int8shiftAlphaAll",
-                "alignment": "soft"
-            }
-            
-            cfg_textyaml = yaml.safe_dump(cfg)   # returns YAML string, no file written
             cfg_text = f'''
             bergamot-mode: wasm
             models:
@@ -176,7 +154,6 @@ class Bergamot:
             gemm-precision: int8shiftAlphaAll
             alignment: soft
             '''
-
             model = self.service.modelFromConfig(cfg_text)
             # model = self.service.modelFromConfigPath(f"{self.localfolder}/config.yml")
             options = bergamot.ResponseOptions(alignment=False, sentenceMappings=False, qualityScores=False, HTML=False)