remove interp frame bg

app.py

@@ -58,7 +58,7 @@ with gr.Blocks() as demo:
     )
 
     interp_btn.click(
-        run_interpolate_frames, inputs=[interp_frame1, interp_frame2, times_to_interp], outputs=[frames, frames_thumb]
+        run_interpolate_frames, inputs=[interp_frame1, interp_frame2, times_to_interp, remove_bg], outputs=[frames, frames_thumb]
     )
 
     eval_btn.click(

main.py

@@ -1293,7 +1293,7 @@ def run_eval(images_orig, video_path, train_steps=100, inference_steps=10, fps=1
 
 
 @spaces.GPU(duration=30)
-def interpolate_frames(frame1, frame2, times_to_interp):
+def interpolate_frames(frame1, frame2, times_to_interp, remove_bg):
     film = Predictor()
     film.setup()
 
@@ -1311,7 +1311,9 @@ def interpolate_frames(frame1, frame2, times_to_interp):
         results = [Image.open(out_vid)]
 
     print(results)
-    
+    if remove_bg:
+        results = [removebg(img, rembg_session, True) for img in results]
+
     for r in results:
         r.thumbnail((width, height))