add example images

app.py

@@ -269,7 +269,6 @@ def main():
 
     # check if there is an input_image
     if not ('input_image' in st.session_state and st.session_state['input_image'] is not None):
-        print("Image not present")
         st.success("Upload an image to start")
         st.write("Welcome to the interior design controlnet demo! "
                  "You can start by uploading a picture of your room, after which you will see "
@@ -293,6 +292,17 @@ def main():
             "so the user can keep elements of their room and change specific parts of the image."
             ""
         )
+        st.write("### Testing images")
+        st.write("If you don't have any pictures close, you can use one of these images to test the model:")
+        st.session_state['example_image_0'] = Image.open("content/example_0.png")
+        st.session_state['example_image_1'] = Image.open("content/example_1.jpg")
+        st.image(st.session_state['example_image_0'], caption="Example image 1", use_column_width=True)
+        if st.button("Use example 1"):
+            move_image('example_image_0', 'initial_image', remove_state=True, rerun=True)
+
+        st.image(st.session_state['example_image_1'], caption="Example image 2", use_column_width=True)
+        if st.button("Use example 2"):
+            move_image('example_image_1', 'initial_image', remove_state=True, rerun=True)
     else:
         make_prompt_row()