Update app.py

app.py

@@ -191,8 +191,16 @@ if __name__ == "__main__":
                 mask_cv = mask_cv.convert("L").resize((384,512))
                 mask_cv = np.array(mask_cv)
                 mask_cv = 255-mask_cv
-    
-                img_C = cv2.seamlessClone(dict, img_cv, mask_cv, (192,256), cv2.NORMAL_CLONE)
+
+                
+                # Normalize mask to binary values
+                _, mask_cv = cv2.threshold(mask_cv, 128, 255, cv2.THRESH_BINARY)
+
+                # Center for seamlessClone
+                center = (target_size[0] // 2, target_size[1] // 2)
+
+                
+                img_C = cv2.seamlessClone(dict, img_cv, mask_cv, center, cv2.NORMAL_CLONE)
                 img_C = cv2.cvtColor(img_C, cv2.COLOR_BGR2RGB)