Update app.py

app.py

@@ -3,10 +3,10 @@ import torch
 import spaces
 import gradio as gr
 from diffusers import FluxFillPipeline
-from diffusers.utils import load_image
 import random
 import numpy as np
 from huggingface_hub import hf_hub_download
+from PIL import Image, ImageOps
 
 
 CSS = """
@@ -65,10 +65,10 @@ def inpaintGen(
     if not mask_path:
         raise gr.Error("Please draw a mask on the image.")
 
-    source_img = load_image(source_path)
-    mask_img = load_image(mask_path)
+    source_img = Image.open(source_path).convert("RGB")
+    mask_img = Image.open(mask_path).convert('L')
 
-    mask_img = create_mask_image(mask_img)
+    #mask_img = create_mask_image(mask_img)
     
     width, height = source_img.size