Update train.py

train.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import os
+import re
 import warnings
 from logging import getLogger
 from multiprocessing import cpu_count
@@ -38,7 +39,7 @@ torch.set_float32_matmul_precision("high")
 
 from pathlib import Path
 
-from huggingface_hub import create_repo, upload_folder, login
+from huggingface_hub import create_repo, upload_folder, login, list_repo_files, delete_file
 
 if os.environ.get("HF_TOKEN"):
     login(os.environ.get("HF_TOKEN"))
@@ -75,7 +76,7 @@ class HuggingFacePushCallback(pl.Callback):
         to_delete = set(repo_ckpts) - set(local_ckpts)
 
         for fname in to_delete:
-            print("🗑 Deleting {fname} from repo")
+            print(f"🗑 Deleting {fname} from repo")
             delete_file(fname, repo_id)