weifeng-chen commited on
Commit
7e31dbb
β€’
1 Parent(s): 2278ab0
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ def resize(w_val,l_val,img):
17
 
18
  def infer(prompt, guide, steps, width, height):
19
  output = pipe_text2img(prompt, guidance_scale=guide, num_inference_steps=steps, width=width, height=height)
20
- image = images.images[0]
21
  return image
22
 
23
  gr.Interface(fn=infer, inputs=
 
17
 
18
  def infer(prompt, guide, steps, width, height):
19
  output = pipe_text2img(prompt, guidance_scale=guide, num_inference_steps=steps, width=width, height=height)
20
+ image = output.images[0]
21
  return image
22
 
23
  gr.Interface(fn=infer, inputs=