b

debug.py

@@ -6,6 +6,7 @@ import time
 from huggingface_hub import CommitScheduler, HfApi
 from PIL import Image
 import numpy as np
+from diffusers.utils import load_image
 
 APP_VERSION = "0_3"
 HF_DATASET_REPO = "LPX55/upscaler_logs"  # Change to your dataset repo
@@ -34,6 +35,11 @@ def save_image(image_id, image_path: Image.Image) -> None:
     print("Image ID Type: " + str(type(image_id)))
     save_image_path = os.path.join(IMAGE_DIR, f"{image_id}")
     print("Save image path: " + save_image_path)
+    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:
             # print("Image path: " + image_path)