Update app.py
Browse files
app.py
CHANGED
|
@@ -44,7 +44,7 @@ def generate(prompt):
|
|
| 44 |
|
| 45 |
def make_image(prompt):
|
| 46 |
pipe = get_image_pipe()
|
| 47 |
-
image = pipe(prompt)
|
| 48 |
|
| 49 |
def predict(prompt):
|
| 50 |
story = generate(prompt=prompt)
|
|
@@ -63,4 +63,4 @@ gr.Interface(
|
|
| 63 |
title=title,
|
| 64 |
description=description,
|
| 65 |
examples=[["My new adventure would be"], ["Then I a hobbit appeared"], ["Frodo told me"]]
|
| 66 |
-
).launch(
|
|
|
|
| 44 |
|
| 45 |
def make_image(prompt):
|
| 46 |
pipe = get_image_pipe()
|
| 47 |
+
image = pipe(prompt)["sample"][0]
|
| 48 |
|
| 49 |
def predict(prompt):
|
| 50 |
story = generate(prompt=prompt)
|
|
|
|
| 63 |
title=title,
|
| 64 |
description=description,
|
| 65 |
examples=[["My new adventure would be"], ["Then I a hobbit appeared"], ["Frodo told me"]]
|
| 66 |
+
).launch(debug=True)
|