fix main

main.py

@@ -15,7 +15,7 @@ try:
 except:
     pass
 
-st.markdown("<h1 style='text-align:center;'> Magic Board</h1>", unsafe_allow_html=True)
+st.markdown("<h1 style='text-align:center;'> Magic Board - CPU version</h1>", unsafe_allow_html=True)
 # Specify canvas parameters in application
 drawing_mode = st.sidebar.selectbox(
     _("Drawing tool:"), ("freedraw", "line", "rect", "circle", "transform")
@@ -64,11 +64,11 @@ if st.button('Submit'):
     if prompt is None:
         st.write("No prompt, no magic! :(")
 
-    with st.spinner('Wait for it...'):
+    with st.spinner('Wait for it... ~ 5 minutes '):
         images = pipe_image(prompt=prompt,
                        init_image=comp_img,
                        strength=strength)
     if images["nsfw_content_detected"]:
         st.write("NSFW content detected, retry with another prompt or image")
-    #for image in images:
-    st.image(images.images)
+    else:
+        st.image(images.images)