fffiloni commited on
Commit
6e43303
1 Parent(s): 7c79805

add progress bar

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ def export_to_video(frames: np.ndarray, fps: int) -> str:
19
  writer.close()
20
  return out_file.name
21
 
22
- def infer(prompt):
23
  negative_prompt = "text, watermark, copyright, blurry, nsfw"
24
  video_frames = pipe(prompt, negative_prompt=negative_prompt, num_inference_steps=40, height=320, width=576, num_frames=24).frames[0]
25
 
 
19
  writer.close()
20
  return out_file.name
21
 
22
+ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
23
  negative_prompt = "text, watermark, copyright, blurry, nsfw"
24
  video_frames = pipe(prompt, negative_prompt=negative_prompt, num_inference_steps=40, height=320, width=576, num_frames=24).frames[0]
25