Update app.py

app.py

@@ -30,7 +30,7 @@ def predict(image, prompt="high quality, best quality", negative_prompt="", guid
     if negative_prompt == "":
         negative_prompt = None
     
-    init_image = image['image'].convert("RGB")    
+    init_image = image['composite'].convert("RGB")    
     # init_image = image.convert("RGB")
     # if center_crop is False:
     # init_image = init_image.resize((224, 224))