not-lain commited on
Commit
d6302a9
1 Parent(s): 6bfcd1c
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -28,10 +28,14 @@ transform_image = transforms.Compose(
28
 
29
 
30
  @spaces.GPU
31
- def fn(vid, fps=12, color="#00FF00"):
32
  # Load the video using moviepy
33
  video = mp.VideoFileClip(vid)
34
 
 
 
 
 
35
  # Extract audio from the video
36
  audio = video.audio
37
 
@@ -94,7 +98,7 @@ with gr.Blocks() as demo:
94
  out_video = gr.Video(label="Final Output Video")
95
  submit_button = gr.Button("Change Background")
96
  with gr.Row():
97
- fps_slider = gr.Slider(minimum=1, maximum=60, step=1, value=12, label="Output FPS")
98
  color_picker = gr.ColorPicker(label="Background Color", value="#00FF00")
99
 
100
  examples = gr.Examples(["rickroll-2sec.mp4"], inputs=in_video, outputs=[stream_image, out_video], fn=fn, cache_examples=True, cache_mode="eager")
 
28
 
29
 
30
  @spaces.GPU
31
+ def fn(vid, fps=0, color="#00FF00"):
32
  # Load the video using moviepy
33
  video = mp.VideoFileClip(vid)
34
 
35
+ # Load original fps if fps value is equal to 0
36
+ if fps == 0:
37
+ fps = video.fps
38
+
39
  # Extract audio from the video
40
  audio = video.audio
41
 
 
98
  out_video = gr.Video(label="Final Output Video")
99
  submit_button = gr.Button("Change Background")
100
  with gr.Row():
101
+ fps_slider = gr.Slider(minimum=0, maximum=60, step=1, value=0, label="Output FPS (0 will inherit the original fps value)")
102
  color_picker = gr.ColorPicker(label="Background Color", value="#00FF00")
103
 
104
  examples = gr.Examples(["rickroll-2sec.mp4"], inputs=in_video, outputs=[stream_image, out_video], fn=fn, cache_examples=True, cache_mode="eager")