fffiloni commited on
Commit
7495fff
1 Parent(s): 9b7b4e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
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,