Spaces:
Paused
Paused
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 |
|