Spaces:
Runtime error
Runtime error
Commit
Β·
7e31dbb
1
Parent(s):
2278ab0
update
Browse files
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 =
|
| 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=
|