debug

app.py

@@ -8,7 +8,6 @@ from kolors.models.tokenization_chatglm import ChatGLMTokenizer
 from diffusers import AutoencoderKL, EulerDiscreteScheduler, UNet2DConditionModel
 import gradio as gr
 import numpy as np
-from PIL import Image
 
 device = "cuda"
 ckpt_dir = snapshot_download(repo_id="Kwai-Kolors/Kolors-Inpainting")
@@ -43,7 +42,7 @@ def infer(prompt,
           num_inference_steps = 25
           ):
     if mask_image != None:
-        image = dict({'background': image, 'layers': [negative_prompt]})
+        image = dict({'background': image, 'layers': [mask_image]})
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator().manual_seed(seed)