Update app.py

app.py

@@ -28,7 +28,7 @@ transform_image = transforms.Compose(
 
 
 @spaces.GPU
-def fn(vid, fps, color):
+def fn(vid, fps=12, color="#00FF00"):
     # Load the video using moviepy
     video = mp.VideoFileClip(vid)
 
@@ -40,6 +40,7 @@ def fn(vid, fps, color):
 
     # Process each frame for background removal
     processed_frames = []
+    yield gr.update(visible=True), gr.update(visible=False)
     for frame in frames:
         pil_image = Image.fromarray(frame)
         processed_image = process(pil_image, color)
@@ -86,15 +87,6 @@ def process(image, color_hex):
     return image
 
 
-def process_file(f, color="#00FF00"):
-    name_path = f.rsplit(".", 1)[0] + ".png"
-    im = load_img(f, output_type="pil")
-    im = im.convert("RGB")
-    transparent = process(im, color)
-    transparent.save(name_path)
-    return name_path
-
-
 with gr.Blocks() as demo:
     with gr.Row():
         in_video = gr.Video(label="Input Video")
@@ -105,13 +97,11 @@ with gr.Blocks() as demo:
         fps_slider = gr.Slider(minimum=1, maximum=60, step=1, value=12, label="Output FPS")
         color_picker = gr.ColorPicker(label="Background Color", value="#00FF00")
 
+    examples = gr.Examples(["rickroll-2sec.mp4"], inputs=in_video, outputs=[stream_image, out_video], fn=fn, cache_examples=True, cache_mode="lazy")
+
     submit_button.click(
-        fn=lambda: gr.update(visible=True), inputs=None, outputs=[stream_image]
-    ).then(
-        fn=lambda: gr.update(visible=False), inputs=None, outputs=[out_video],
-    ).then(
         fn, inputs=[in_video, fps_slider, color_picker], outputs=[stream_image, out_video]
     )
-
+    
 if __name__ == "__main__":
     demo.launch(show_error=True)