DingoBeast commited on
Commit
2484b4f
·
verified ·
1 Parent(s): 7338c50

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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=50).images[0]
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)