Spaces:
Running
on
A10G
Running
on
A10G
add progress bar
Browse files
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 |
|