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"{img_id}"),
+        path_or_fileobj=os.path.join("/tmp/gradio", f"{img_id}"),
 
         path_in_repo="/v" + APP_VERSION + "/" + img_id,
         repo_id=HF_DATASET_REPO,
@@ -44,15 +44,17 @@ def save_image(image_id, image_path: Image.Image) -> None:
     print("Save image path: " + save_image_path)
     try:
         loaded = load_image(image_path)
+        cache_file = "/tmp/gradio/" + str(uuid.uuid4()) + ".png"
+        loaded.save(cache_file, "PNG")
         print("Loaded Type: " + str(type(loaded)))
         with scheduler.lock:
             try:
-                cache_temp(save_image_path)
-                # print("Image path: " + image_path)
-                print("Type: " + str(type(image_path)))
+                cache_temp(cache_file)
+                print("Cache path: " + cache_file)
+                print("Type: " + str(type(cache_file)))
                 try:
-                    # img2 = img2.open(img)
-                    loaded.save(save_image_path)
+                    img2 = Image.open(cache_file)
+                    img2.save(save_image_path)
                     
                     print(f"Saved image: {save_image_path}")
                 except Exception as e: