Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -141,9 +141,9 @@ def run_inference(prompt, video_path, condition, video_length, seed):
|
|
| 141 |
|
| 142 |
print(f"RUNNING INFERENCE ...")
|
| 143 |
if video_length > 12:
|
| 144 |
-
command = f"python inference.py --prompt '{prompt}' --condition '{condition}' --video_path '{normalized}' --output_path '{output_path}' --temp_chunk_path 'result' --width 512 --height 512 --fps {target_fps} --seed {seed} --video_length {video_length} --is_long_video"
|
| 145 |
else:
|
| 146 |
-
command = f"python inference.py --prompt '{prompt}' --condition '{condition}' --video_path '{normalized}' --output_path '{output_path}' --temp_chunk_path 'result' --width 512 --height 512 --fps {target_fps} --seed {seed} --video_length {video_length}"
|
| 147 |
subprocess.run(command, shell=True)
|
| 148 |
|
| 149 |
# Construct the video path
|
|
@@ -183,7 +183,8 @@ with gr.Blocks(css=css) as demo:
|
|
| 183 |
status = gr.Textbox(label="result")
|
| 184 |
video_path.change(fn=get_frame_count,
|
| 185 |
inputs=[video_path],
|
| 186 |
-
outputs=[video_length]
|
|
|
|
| 187 |
)
|
| 188 |
submit_btn.click(fn=run_inference,
|
| 189 |
inputs=[prompt,
|
|
|
|
| 141 |
|
| 142 |
print(f"RUNNING INFERENCE ...")
|
| 143 |
if video_length > 12:
|
| 144 |
+
command = f"python inference.py --prompt '{prompt}' --condition '{condition}' --video_path '{normalized}' --output_path '{output_path}' --temp_chunk_path 'result' --width 512 --height 512 --fps {target_fps} --seed {seed} --video_length {video_length} --smoother_steps 10 12 --is_long_video"
|
| 145 |
else:
|
| 146 |
+
command = f"python inference.py --prompt '{prompt}' --condition '{condition}' --video_path '{normalized}' --output_path '{output_path}' --temp_chunk_path 'result' --width 512 --height 512 --fps {target_fps} --seed {seed} --video_length {video_length} --smoother_steps 19 20"
|
| 147 |
subprocess.run(command, shell=True)
|
| 148 |
|
| 149 |
# Construct the video path
|
|
|
|
| 183 |
status = gr.Textbox(label="result")
|
| 184 |
video_path.change(fn=get_frame_count,
|
| 185 |
inputs=[video_path],
|
| 186 |
+
outputs=[video_length],
|
| 187 |
+
queue=False
|
| 188 |
)
|
| 189 |
submit_btn.click(fn=run_inference,
|
| 190 |
inputs=[prompt,
|