Update app.py

app.py

@@ -13,7 +13,7 @@ OUTPUT_DIR = "outputs"
 os.makedirs(OUTPUT_DIR, exist_ok=True)
 
 # --- DEVICE SETUP ---
-device = "cpu"  # Force CPU for safety
+device = "cpu"  # Force CPU
 
 # --- LOAD STABLE DIFFUSION PIPELINE ---
 MODEL_NAME = "Lykon/anything-cartoon"  # original model
@@ -38,9 +38,13 @@ pipe.to(device)
 
 # --- HELPER FUNCTIONS ---
 def load_image(file):
-    ext = os.path.splitext(file.name)[1].lower()
+    ext = os.path.splitext(file.name)[1].lower() if hasattr(file, "name") else os.path.splitext(file)[1].lower()
     if ext in [".heic", ".heif"]:
-        heif_file = pyheif.read_heif(file.read())
+        if hasattr(file, "read"):
+            heif_file = pyheif.read_heif(file.read())
+        else:
+            with open(file, "rb") as f:
+                heif_file = pyheif.read_heif(f.read())
         image = Image.frombytes(
             heif_file.mode,
             heif_file.size,
@@ -51,8 +55,11 @@ def load_image(file):
         )
         return image
     else:
-        file.seek(0)
-        return Image.open(file)
+        if hasattr(file, "seek"):
+            file.seek(0)
+            return Image.open(file)
+        else:
+            return Image.open(file)
 
 def apply_filters(file, mode):
     image = load_image(file)
@@ -94,7 +101,7 @@ def apply_filters(file, mode):
     elif mode == "AI Cartoon":
         pil_img = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
         prompt = "cartoon, anime style, vibrant colors, detailed, smooth"
-        # CPU only, no GPU/autocast
+        # CPU only
         cartoon_img = pipe(prompt=prompt, image=pil_img, num_inference_steps=30).images[0]
         processed = np.array(cartoon_img)
     elif mode == "Sepia":
@@ -124,7 +131,7 @@ def apply_filters(file, mode):
 demo = gr.Interface(
     fn=apply_filters,
     inputs=[
-        gr.File(label="Upload Image (PNG, JPG, HEIC)", type="file"),
+        gr.File(label="Upload Image (PNG, JPG, HEIC)", type="filepath"),  # FIXED
         gr.Radio(
             choices=["Gray", "Scratch", "Pencil Sketch", "Cartoon", "AI Cartoon",
                      "Sepia", "Edge Detection", "RGB", "HSV"],