fix main, add few writes

main.py

@@ -59,15 +59,17 @@ if st.button('Submit'):
         comp_img = Image.new("RGB", img_size)
         comp_img = Image.blend(img_draw, img_back, alpha=0.5)
     else:
-        raise "Upload an image or draw something"
+        st.write("Upload an image or draw something")
     
     if prompt is None:
-        raise "No prompt, no magic! :("
+        st.write("No prompt, no magic! :(")
 
     with st.spinner('Wait for it...'):
         images = pipe_image(prompt=prompt,
                        init_image=comp_img,
                        strength=strength)
     st.success("Image is ready!")
+    if len(images["nsfw_content_detected"]) > 0:
+        st.write("NSFW content detected, retry with another prompt or image")
     for image in images:
         st.image(image)