Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
cbadedf
1
Parent(s):
77eb8e2
progress-bar
Browse files
app.py
CHANGED
|
@@ -96,6 +96,7 @@ def generate(
|
|
| 96 |
num_inference_steps_base: int = 25,
|
| 97 |
num_inference_steps_refiner: int = 25,
|
| 98 |
apply_refiner: bool = False,
|
|
|
|
| 99 |
) -> PIL.Image.Image:
|
| 100 |
print(f"** Generating image for: \"{prompt}\" **")
|
| 101 |
generator = torch.Generator().manual_seed(seed)
|
|
|
|
| 96 |
num_inference_steps_base: int = 25,
|
| 97 |
num_inference_steps_refiner: int = 25,
|
| 98 |
apply_refiner: bool = False,
|
| 99 |
+
progress=gr.Progress(track_tqdm=True),
|
| 100 |
) -> PIL.Image.Image:
|
| 101 |
print(f"** Generating image for: \"{prompt}\" **")
|
| 102 |
generator = torch.Generator().manual_seed(seed)
|