trashchenkov commited on
Commit
acd8555
·
verified ·
1 Parent(s): 98a5d61

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -40,7 +40,7 @@ def infer(
40
  height,
41
  guidance_scale,
42
  num_inference_steps,
43
- progress=gr.Progress(track_tqdm=True),
44
  ):
45
  global pipe
46
 
@@ -60,7 +60,7 @@ def infer(
60
  prompt=prompt,
61
  negative_prompt=negative_prompt,
62
  guidance_scale=guidance_scale,
63
- num_inference_steps=num_inference_steps,
64
  width=width,
65
  height=height,
66
  generator=generator,
@@ -167,4 +167,4 @@ with gr.Blocks(css=css) as demo:
167
  )
168
 
169
  if __name__ == "__main__":
170
- demo.launch()
 
40
  height,
41
  guidance_scale,
42
  num_inference_steps,
43
+ progress=None,
44
  ):
45
  global pipe
46
 
 
60
  prompt=prompt,
61
  negative_prompt=negative_prompt,
62
  guidance_scale=guidance_scale,
63
+ num_inference_steps=int(num_inference_steps), # Ensure it's an integer
64
  width=width,
65
  height=height,
66
  generator=generator,
 
167
  )
168
 
169
  if __name__ == "__main__":
170
+ demo.launch()