Update debug.py

debug.py

@@ -16,8 +16,7 @@ LOG_DIR = "logs_" + APP_VERSION
 IMAGE_DIR = os.path.join(LOG_DIR, "upscaler")
 
 LOG_FILE = os.path.join(LOG_DIR, f"{int(time.time())}-logs.csv")
-
-
+api = HfApi(token=HF_TOKEN)
 scheduler = CommitScheduler(
     repo_id=HF_DATASET_REPO,
     repo_type="dataset",
@@ -26,9 +25,17 @@ scheduler = CommitScheduler(
     private=True,
     token=HF_TOKEN,
     path_in_repo="v" + APP_VERSION
-
 )
 
+def cache_temp(img_id):
+    api.upload_file(
+        path_or_fileobj="/path/to/local/folder/README.md",
+
+        path_in_repo="/v" + APP_VERSION + "/" + img_id,
+        repo_id=HF_DATASET_REPO,
+        repo_type="dataset",
+        token=HF_TOKEN,
+    )
 def save_image(image_id, image_path: Image.Image) -> None:
     os.makedirs(IMAGE_DIR, exist_ok=True)
     print("Image ID: " + image_id)
@@ -42,6 +49,7 @@ def save_image(image_id, image_path: Image.Image) -> None:
         print(f"Error loading image: {str(e)}")
     with scheduler.lock:
         try:
+            cache_temp(save_image_path)
             # print("Image path: " + image_path)
             print("Type: " + str(type(image_path)))
             with load_image(image_path) as img: