fix main and utils

main.py

@@ -68,8 +68,7 @@ if st.button('Submit'):
         images = pipe_image(prompt=prompt,
                        init_image=comp_img,
                        strength=strength)
-    st.success("Image is ready!")
-    if images["nsfw_content_detected"][0]:
+    if images["nsfw_content_detected"]:
         st.write("NSFW content detected, retry with another prompt or image")
     for image in images:
         st.image(image)

utils.py

@@ -21,5 +21,5 @@ def pipe_image(prompt,
                     init_image=init_image,
                     strength=strength,
                     guidance_scale=7.5,
-                    num_inference_steps=50).images 
+                    num_inference_steps=50)
     return images