add content

.gitattributes

@@ -37,3 +37,7 @@ content/inpainting_after.png filter=lfs diff=lfs merge=lfs -text
 content/inpainting_before.jpg filter=lfs diff=lfs merge=lfs -text
 content/inpainting_sidebar.png filter=lfs diff=lfs merge=lfs -text
 content/regen_example.png filter=lfs diff=lfs merge=lfs -text
+content/example_0.png filter=lfs diff=lfs merge=lfs -text
+content/example_1.jpg filter=lfs diff=lfs merge=lfs -text
+content/output_0.png filter=lfs diff=lfs merge=lfs -text
+content/output_1.png filter=lfs diff=lfs merge=lfs -text

app.py

@@ -294,15 +294,30 @@ def main():
         )
         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)
+        
+        col_im_0, col_im_1 = st.columns(2)
+        
+        with col_im_0:
+            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', 'input_image', remove_state=True, rerun=False)
+                move_image('example_image_0', 'initial_image', remove_state=True, rerun=True)
+        with col_im_1:
+            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', 'input_image', remove_state=True, rerun=False)
+                move_image('example_image_1', 'initial_image', remove_state=True, rerun=True)
+        
+        st.write("## Generated examples")
+        col_ex_0, col_ex_1 = st.columns(2)
+        with col_ex_0:
+            st.image(Image.open("content/output_1.png"), caption="Generated example 1, regenerating certain objects in the room", use_column_width=True)
+        with col_ex_1:
+            st.image(Image.open("content/output_0.png"), caption="Generated example 1, regenerating certain objects in the room", use_column_width=True)
+            
     else:
         make_prompt_row()