Spaces:

yuvaranianandhan24
/

story_telling

Sleeping

yuvaranianandhan24 commited on Apr 24, 2024

Commit

19a195c

verified ·

1 Parent(s): 38224de

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -122,7 +122,7 @@ def image_generator_page():
     if st.button("Generate Image"):
         if prompt:
             with st.spinner("Generating image..."):
-                image = txt2img(prompt=prompt, style=image_style)
             st.image(image)
         else:
             st.error("Please enter a prompt...")

     if st.button("Generate Image"):
         if prompt:
             with st.spinner("Generating image..."):
+                image = txt2img(prompt, style=image_style)
             st.image(image)
         else:
             st.error("Please enter a prompt...")