add background to output

app.py

@@ -35,10 +35,7 @@ def run_front_detection(input_img):
     image_rgb = image_gray.convert('RGB')
     np_array = np.asarray(image_rgb).copy()
 
-    # Modify the NumPy array (e.g. change pixel values)
-    # Example: multiply all pixel values by 2
-
-    np_array[front==1,0]= 255
+    np_array[np.array(front)==255,0] = 255
 
     # Convert NumPy array back to PIL image
     pil_image_modified = Image.fromarray(np_array)