Update app.py

app.py

@@ -108,7 +108,7 @@ class Bergamot:
                 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:
+                    if self.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)
@@ -124,7 +124,7 @@ class Bergamot:
                 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)
+                    self.installed_pairs.add(self.localfolder)
             except Exception as dryrunerror:
                 print('installed_pairs', self.installed_pairs, 'localfolder', self.localfolder, 'dry_runerror', dryrunerror)
         except Exception as downloaderror:
@@ -134,7 +134,7 @@ class Bergamot:
         self.downloadbergamotfiles()
         try:
             model = self.service.modelFromConfigPath(f"{self.localfolder}/config.yml")
-            # model = self.service.modelFromConfig(localfolder)
+            # model = self.service.modelFromConfig(self.localfolder)
             options = bergamot.ResponseOptions(alignment=False, sentenceMappings=False, qualityScores=False, HTML=False)
             rawresponse = self.service.translate(model, bergamot.VectorString(self.input_text), options)
             response: list|str = [r.target.text for r in rawresponse] if len(rawresponse) > 1 else next(iter(rawresponse)).target.text