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, steps):
|
|
| 141 |
|
| 142 |
print(f"RUNNING INFERENCE ...")
|
| 143 |
if video_length > 12:
|
| 144 |
-
command = f"python inference.py --prompt '{prompt}' --inference_steps {steps} --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
|
| 145 |
else:
|
| 146 |
-
command = f"python inference.py --prompt '{prompt}' --inference_steps {steps} --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 |
try:
|
| 148 |
subprocess.run(command, shell=True)
|
| 149 |
|
|
@@ -157,7 +157,7 @@ def run_inference(prompt, video_path, condition, video_length, seed, steps):
|
|
| 157 |
|
| 158 |
print(f"FINISHED !")
|
| 159 |
return "done", 'resized_final.mp4'
|
| 160 |
-
except:
|
| 161 |
return "error", None
|
| 162 |
|
| 163 |
|
|
@@ -180,7 +180,7 @@ with gr.Blocks(css=css) as demo:
|
|
| 180 |
with gr.Row():
|
| 181 |
condition = gr.Dropdown(label="Condition", choices=["depth", "canny", "pose"], value="depth")
|
| 182 |
seed = gr.Number(label="seed", value=42)
|
| 183 |
-
inference_steps = gr.Slider(label="Inference steps", minimum=
|
| 184 |
submit_btn = gr.Button("Submit")
|
| 185 |
with gr.Column():
|
| 186 |
video_res = gr.Video(label="result")
|
|
|
|
| 141 |
|
| 142 |
print(f"RUNNING INFERENCE ...")
|
| 143 |
if video_length > 12:
|
| 144 |
+
command = f"python inference.py --prompt '{prompt}' --inference_steps {steps} --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] --is_long_video"
|
| 145 |
else:
|
| 146 |
+
command = f"python inference.py --prompt '{prompt}' --inference_steps {steps} --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 |
try:
|
| 148 |
subprocess.run(command, shell=True)
|
| 149 |
|
|
|
|
| 157 |
|
| 158 |
print(f"FINISHED !")
|
| 159 |
return "done", 'resized_final.mp4'
|
| 160 |
+
except RuntimeError:
|
| 161 |
return "error", None
|
| 162 |
|
| 163 |
|
|
|
|
| 180 |
with gr.Row():
|
| 181 |
condition = gr.Dropdown(label="Condition", choices=["depth", "canny", "pose"], value="depth")
|
| 182 |
seed = gr.Number(label="seed", value=42)
|
| 183 |
+
inference_steps = gr.Slider(label="Inference steps", minimum=25, maximum=50, step=1, value=25)
|
| 184 |
submit_btn = gr.Button("Submit")
|
| 185 |
with gr.Column():
|
| 186 |
video_res = gr.Video(label="result")
|