Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -15,13 +15,10 @@ def predict(prompt):
|
|
15 |
""" this function accepts a prompt and generate a
|
16 |
corresponding image for the given prompt """
|
17 |
|
18 |
-
image = pipeline(prompt=prompt)
|
19 |
|
20 |
return image
|
21 |
|
22 |
-
image = predict(" a man that wears a cat")
|
23 |
-
image.show()
|
24 |
-
|
25 |
### 4. Gradio app ###
|
26 |
|
27 |
# Create title, description and article strings
|
|
|
15 |
""" this function accepts a prompt and generate a
|
16 |
corresponding image for the given prompt """
|
17 |
|
18 |
+
image = pipeline(prompt=prompt).images[0]
|
19 |
|
20 |
return image
|
21 |
|
|
|
|
|
|
|
22 |
### 4. Gradio app ###
|
23 |
|
24 |
# Create title, description and article strings
|