HelloSun commited on
Commit
8f7841a
·
verified ·
1 Parent(s): 2e1dba8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -48,8 +48,8 @@ def infer(prompt, num_inference_steps):
48
  #negative_prompt = negative_prompt,
49
  guidance_scale = 7.0,
50
  num_inference_steps = num_inference_steps,
51
- width = width,
52
- height = height,
53
  num_images_per_prompt=num_images,
54
  ).images[0]
55
 
 
48
  #negative_prompt = negative_prompt,
49
  guidance_scale = 7.0,
50
  num_inference_steps = num_inference_steps,
51
+ width = 512,
52
+ height = 512,
53
  num_images_per_prompt=num_images,
54
  ).images[0]
55