HelloSun commited on
Commit
6694cc8
·
verified ·
1 Parent(s): 8f7841a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -50,7 +50,7 @@ def infer(prompt, num_inference_steps):
50
  num_inference_steps = num_inference_steps,
51
  width = 512,
52
  height = 512,
53
- num_images_per_prompt=num_images,
54
  ).images[0]
55
 
56
  return image
 
50
  num_inference_steps = num_inference_steps,
51
  width = 512,
52
  height = 512,
53
+ num_images_per_prompt=1,
54
  ).images[0]
55
 
56
  return image