Update app.py
Browse files
app.py
CHANGED
|
@@ -64,7 +64,7 @@ NUM_INFERENCE_STEPS = 4
|
|
| 64 |
|
| 65 |
|
| 66 |
@spaces.GPU
|
| 67 |
-
def infer(prompt, seed, randomize_seed, guidance_scale, num_inference_steps, negative_prompt):
|
| 68 |
if randomize_seed:
|
| 69 |
seed = random.randint(0, MAX_SEED)
|
| 70 |
|
|
|
|
| 64 |
|
| 65 |
|
| 66 |
@spaces.GPU
|
| 67 |
+
def infer(prompt, seed, randomize_seed, guidance_scale, num_inference_steps, negative_prompt, progress=gr.Progress(track_tqdm=True)):
|
| 68 |
if randomize_seed:
|
| 69 |
seed = random.randint(0, MAX_SEED)
|
| 70 |
|