Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
f202d56
1
Parent(s):
5d635cf
add inference
Browse files
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 |
|