Spaces:
Running
on
Zero
Running
on
Zero
add progress tracking
Browse files
app.py
CHANGED
|
@@ -54,7 +54,7 @@ def resize_image(input_path, output_path, target_height):
|
|
| 54 |
|
| 55 |
return output_path, new_width, target_height
|
| 56 |
|
| 57 |
-
def infer(image_in, prompt, inference_steps, guidance_scale, control_weight):
|
| 58 |
|
| 59 |
n_prompt = 'NSFW, nude, naked, porn, ugly'
|
| 60 |
|
|
|
|
| 54 |
|
| 55 |
return output_path, new_width, target_height
|
| 56 |
|
| 57 |
+
def infer(image_in, prompt, inference_steps, guidance_scale, control_weight, progress=gr.Progress(track_tqdm=True)):
|
| 58 |
|
| 59 |
n_prompt = 'NSFW, nude, naked, porn, ugly'
|
| 60 |
|