coralLight commited on
Commit
f202d56
·
1 Parent(s): 5d635cf

add inference

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -251,7 +251,7 @@ def infer(
251
  ,num_inference_steps=20
252
  ,guidance_scale=guidance_scale
253
  ,height=height
254
- ,width=width).images
255
  return image_quick, image_50_steps, seed
256
 
257
 
 
251
  ,num_inference_steps=20
252
  ,guidance_scale=guidance_scale
253
  ,height=height
254
+ ,width=width).images[0]
255
  return image_quick, image_50_steps, seed
256
 
257