Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -69,7 +69,7 @@ def generate_prompt(emotions):
|
|
69 |
return "Create a multi-style artwork combining " + ", ".join(prompt_parts)
|
70 |
|
71 |
def generate_image(prompt):
|
72 |
-
image = text2img(prompt, num_inference_steps=
|
73 |
return image
|
74 |
|
75 |
def main():
|
@@ -87,8 +87,8 @@ def main():
|
|
87 |
st.write(f"- {emotion}")
|
88 |
|
89 |
prompt = generate_prompt(emotions)
|
90 |
-
st.write("Generated prompt:")
|
91 |
-
st.write(prompt)
|
92 |
|
93 |
with st.spinner("Generating painting... This may take a while."):
|
94 |
image = generate_image(prompt)
|
|
|
69 |
return "Create a multi-style artwork combining " + ", ".join(prompt_parts)
|
70 |
|
71 |
def generate_image(prompt):
|
72 |
+
image = text2img(prompt, num_inference_steps=10).images[0]
|
73 |
return image
|
74 |
|
75 |
def main():
|
|
|
87 |
st.write(f"- {emotion}")
|
88 |
|
89 |
prompt = generate_prompt(emotions)
|
90 |
+
#st.write("Generated prompt:")
|
91 |
+
#st.write(prompt)
|
92 |
|
93 |
with st.spinner("Generating painting... This may take a while."):
|
94 |
image = generate_image(prompt)
|