Update app.py

app.py

@@ -67,7 +67,6 @@ def make_input_fields():
     
     return guidance_scale, inference_steps, generator_seed, static_latents, latent_walk
 
-
 def decode_image(image):
     cv2_img = cv2.imdecode(np.frombuffer(image, np.uint8), cv2.IMREAD_COLOR)
     cv2_img = cv2.cvtColor(cv2_img, cv2.COLOR_BGR2RGB)
@@ -95,9 +94,10 @@ if __name__ == "__main__":
         canvas = make_canvas(image)
     
     if st.button("Inpaint"):
-        st.write("Inpainting!")
-        mask_image = get_mask(np.array(canvas.image_data))
+        st.write("Start inpainting process")
+        mask = get_mask(np.array(canvas.image_data))
         result = inpainting(image, mask_image, prompt, negative_prompt)
+        st.session_state["result"] = result
     else:
         result = None
 
@@ -106,5 +106,5 @@ if __name__ == "__main__":
         st.write("The generated image will appear here.")
         if webcam:
             st.image(webcam)
-        if result:
-            st.image(result)
+        if 'results' in st.session_state:
+            st.image(st.session_state["result"])