Update app.py

app.py

@@ -130,15 +130,16 @@ def bergamot(input_text: list[str] = Query(description="Input string or list of
                     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
         ## Check if model/localfolder in repo files, add to set if exists
-        modelcheck = [i for i in all_files if localfolder in i]
+        modelcheck = [i for i in all_files if model_name in i]
+        print('Modelcheck', modelcheck)
         if modelcheck:
-            print('Add to set after modelcheck')
+            print('Add to set after modelcheck', 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', type(dry_run), dry_run)
             if isinstance(dry_run, str):
-                print('Add to set after dryrun')
+                print('Add to set after dryrun', dry_run)
                 installed_pairs.add(localfolder)
         except Exception as dryrunerror:
             print('installed_pairs', installed_pairs, 'localfolder', localfolder, 'dry_runerror', dryrunerror)