fix utils and main

main.py

@@ -63,10 +63,11 @@ if st.button('Submit'):
     
     if prompt is None:
         raise "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!")
     for image in images:
         st.image(image)

utils.py

@@ -15,7 +15,7 @@ def pipe_image(prompt,
                                                 torch_dtype=torch.float32,
                                                 use_auth_token=TOKEN
                                                 )
-    pipe = pipe.to(device)
+    pipe.to(device)
     pipe.enable_attention_slicing()
     images = pipe(prompt=prompt,
                     init_image=init_image,