Update app.py

app.py

@@ -73,13 +73,19 @@ def predict(dict, prompt="", negative_prompt="", guidance_scale=7.5, steps=30, s
         strength=strength,
         clip_skip=1
     )
-    print("Positive:",prompt)
-    print("Negative:",negative_prompt)
-    print("Guidance_scale:",guidance_scale)
-    print("Steps:",steps)
-    print("Strength:",strength)
-    print("Scheduler:",scheduler)
-    return output.images[0].resize(init_image.size, resample=Image.LANCZOS), gr.update(visible=True)
+    inpainted_image = output.images[0]
+
+    # Combine the original and inpainted images using the mask
+    combined_image = Image.composite(inpainted_image, init_image, mask_image.split()[3])
+
+    print("Positive:", prompt)
+    print("Negative:", negative_prompt)
+    print("Guidance_scale:", guidance_scale)
+    print("Steps:", steps)
+    print("Strength:", strength)
+    print("Scheduler:", scheduler)
+
+    return combined_image.resize(init_image.size, resample=Image.LANCZOS), gr.update(visible=True)
 
 
 css = '''