Update app.py

app.py

@@ -107,7 +107,6 @@ def bergamot(input_text: list[str] = Query(description="Input string or list of
     """
     try:
         import bergamot
-        # local_path = ''
         # input_text = [input_text] if isinstance(input_text, str) else input_text           
         config = bergamot.ServiceConfig(numWorkers=4)
         service = bergamot.Service(config)
@@ -130,12 +129,17 @@ def bergamot(input_text: list[str] = Query(description="Input string or list of
                     # 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=repo_id, subfolder=model_name, filename=file_path, local_dir=subfolder)
                     print(f"Downloaded to: {local_path}") # Downloaded to: deen/base/deen/config.yml
-        # if local_path and isinstance(local_path, str):
-        if localfolder in all_files:
+        ## Check if model/localfolder in repo files, add to set if exists
+        modelcheck = [i for i in all_files if localfolder in i]
+        if modelcheck:
+            print('Add to set after modelcheck')
             installed_pairs.add(localfolder)
         try:
             dry_run = hf_hub_download(repo_id=repo_id, subfolder=model_name, filename='config.yml', local_dir=subfolder)
-            print('installed_pairs', installed_pairs, 'localfolder', localfolder, 'dry_run', dry_run)
+            print('installed_pairs', installed_pairs, 'localfolder', localfolder, 'dry_run', type(dry_run), dry_run)
+            if isinstance(dry_run, str):
+                print('Add to set after dryrun')
+                installed_pairs.add(localfolder)
         except Exception as dryrunerror:
             print('installed_pairs', installed_pairs, 'localfolder', localfolder, 'dry_runerror', dryrunerror)
         model = service.modelFromConfigPath(f"{localfolder}/config.yml")