Spaces:
Running
on
Zero
Running
on
Zero
gokaygokay
commited on
Commit
•
6f165b5
1
Parent(s):
72fd6ee
Update app.py
Browse files
app.py
CHANGED
@@ -275,7 +275,7 @@ def frames_to_video(input_folder, output_path, fps, original_video_path):
|
|
275 |
os.remove(temp_output_path)
|
276 |
|
277 |
@timer_func
|
278 |
-
def process_video(input_video, resolution, num_inference_steps, strength, hdr, guidance_scale, max_frames=None, frame_interval=1, preserve_frames=False, batch_size=
|
279 |
abort_event.clear() # Clear the abort flag at the start of a new job
|
280 |
print("Starting video processing...")
|
281 |
model_manager.load_models(progress) # Ensure models are loaded
|
@@ -427,7 +427,7 @@ with gr.Blocks(css=css, theme=gr.themes.Default(primary_hue="blue")) as iface:
|
|
427 |
max_frames = gr.Number(label="Max Frames to Process (leave empty for full video)", precision=0)
|
428 |
frame_interval = gr.Slider(1, 30, 1, step=1, label="Frame Interval (process every nth frame)")
|
429 |
preserve_frames = gr.Checkbox(label="Preserve Existing Processed Frames", value=True)
|
430 |
-
batch_size = gr.Slider(1, 16,
|
431 |
|
432 |
with gr.Column(scale=1):
|
433 |
submit_button = gr.Button("Process Media")
|
|
|
275 |
os.remove(temp_output_path)
|
276 |
|
277 |
@timer_func
|
278 |
+
def process_video(input_video, resolution, num_inference_steps, strength, hdr, guidance_scale, max_frames=None, frame_interval=1, preserve_frames=False, batch_size=8, progress=gr.Progress()):
|
279 |
abort_event.clear() # Clear the abort flag at the start of a new job
|
280 |
print("Starting video processing...")
|
281 |
model_manager.load_models(progress) # Ensure models are loaded
|
|
|
427 |
max_frames = gr.Number(label="Max Frames to Process (leave empty for full video)", precision=0)
|
428 |
frame_interval = gr.Slider(1, 30, 1, step=1, label="Frame Interval (process every nth frame)")
|
429 |
preserve_frames = gr.Checkbox(label="Preserve Existing Processed Frames", value=True)
|
430 |
+
batch_size = gr.Slider(1, 16, 8, step=1, label="Batch Size")
|
431 |
|
432 |
with gr.Column(scale=1):
|
433 |
submit_button = gr.Button("Process Media")
|