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)
|