Update debug.py

debug.py

@@ -29,7 +29,7 @@ scheduler = CommitScheduler(
 
 def cache_temp(img_id):
     api.upload_file(
-        path_or_fileobj=os.path.join("/tmp/logs_0_4/upscaler", f"{image_id}"),
+        path_or_fileobj=os.path.join("/tmp/logs_0_4/upscaler", f"{img_id}"),
 
         path_in_repo="/v" + APP_VERSION + "/" + img_id,
         repo_id=HF_DATASET_REPO,
@@ -45,23 +45,24 @@ def save_image(image_id, image_path: Image.Image) -> None:
     try:
         loaded = load_image(image_path)
         print("Loaded Type: " + str(type(loaded)))
-    except Exception as e:
-        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 scheduler.lock:
             try:
-                # img2 = img2.open(img)
-                loaded.save(save_image_path)
-                
-                print(f"Saved image: {save_image_path}")
+                cache_temp(save_image_path)
+                # print("Image path: " + image_path)
+                print("Type: " + str(type(image_path)))
+                try:
+                    # img2 = img2.open(img)
+                    loaded.save(save_image_path)
+                    
+                    print(f"Saved image: {save_image_path}")
+                except Exception as e:
+                    print(f"Error saving image: {str(e)}")
+    
             except Exception as e:
                 print(f"Error saving image: {str(e)}")
 
-        except Exception as e:
-            print(f"Error saving image: {str(e)}")
+    except Exception as e:
+        print(f"Error loading image: {str(e)}")
 
 def log_params(
     prompt, scale, steps, controlnet_conditioning_scale, guidance_scale, seed, guidance_end,