Update app.py

app.py

@@ -43,13 +43,9 @@ def fn(vid, fps, color):
  processed_frames_changed_bg = []
  for frame in frames:
  pil_image = Image.fromarray(frame)
- processed_image, mask = process(pil_image, color) # Get both processed image and mask
- processed_frames_no_bg.append(np.array(processed_image)) # Save no-background frame
- 
- # Compose with background for changed background video
- background = Image.new("RGBA", pil_image.size, color + str((255,)))
- composed_image = Image.composite(pil_image, background, mask)
- processed_frames_changed_bg.append(np.array(composed_image))
+ processed_image, mask = process(pil_image, color) 
+ processed_frames_no_bg.append(np.array(mask)) 
+ processed_frames_changed_bg.append(np.array(processed_image))
 
  # Create a new video from the processed frames
  processed_video = mp.ImageSequenceClip(processed_frames_changed_bg, fps=fps)