Update app.py

app.py

@@ -132,6 +132,7 @@ def floorplan_generation(outline, num_of_rooms):
         mask = np.array(n_outline)
         mask = cv2.cvtColor(mask, cv2.COLOR_RGB2BGR)
         image_c = clean_img(image, mask)
+        image_c = Image.fromarray(image_c).convert('RGB')
         
         mask_arr = np.array(image_c)
         mask_bgr = cv2.cvtColor(mask_arr, cv2.COLOR_RGB2BGR)