Update app.py
Browse files
app.py
CHANGED
@@ -118,7 +118,7 @@ def infer(video_in, trim_value, prompt,
|
|
118 |
break_vid = get_frames(video_in)
|
119 |
frames_list= break_vid[0]
|
120 |
fps = break_vid[1]
|
121 |
-
n_frame = int(trim_value
|
122 |
#n_frame = len(frames_list)
|
123 |
|
124 |
if n_frame >= len(frames_list):
|
@@ -218,7 +218,7 @@ with gr.Blocks() as demo:
|
|
218 |
submit_btn.click(
|
219 |
fn=infer,
|
220 |
inputs=[
|
221 |
-
video_in,
|
222 |
],
|
223 |
outputs=output
|
224 |
)
|
|
|
118 |
break_vid = get_frames(video_in)
|
119 |
frames_list= break_vid[0]
|
120 |
fps = break_vid[1]
|
121 |
+
n_frame = int(trim_value * fps)
|
122 |
#n_frame = len(frames_list)
|
123 |
|
124 |
if n_frame >= len(frames_list):
|
|
|
218 |
submit_btn.click(
|
219 |
fn=infer,
|
220 |
inputs=[
|
221 |
+
video_in, trim_in, prompt, negative_prompt, num_inference_steps, size, guidance_scale, seed
|
222 |
],
|
223 |
outputs=output
|
224 |
)
|