example

.gitattributes

@@ -50,3 +50,4 @@ content/keep[[:space:]]background[[:space:]]1.png filter=lfs diff=lfs merge=lfs
 content/keep[[:space:]]background[[:space:]]2.png filter=lfs diff=lfs merge=lfs -text
 content/output_2.png filter=lfs diff=lfs merge=lfs -text
 content/example_3.jpg filter=lfs diff=lfs merge=lfs -text
+content/output_3.png filter=lfs diff=lfs merge=lfs -text

app.py

@@ -29,6 +29,13 @@ def on_upload() -> None:
         if 'output_image' in st.session_state:
             del st.session_state['output_image']
 
+def make_image_row(image_0, image_1):
+    col_0, col_1 = st.columns(2)
+    with col_0:
+        st.image(image_0, use_column_width=True)
+    with col_1:
+        st.image(image_1, use_column_width=True)
+
 
 def check_reset_state() -> bool:
     """Check whether the UI elements need to be reset
@@ -345,15 +352,10 @@ def main():
                 move_image('example_image_3', '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, regenerating certain objects in the room", use_column_width=True)
-            st.image(Image.open("content/regen_example.png"), caption="Generated example, regenerating certain objects in the room", use_column_width=True)
-            st.image(Image.open("content/keep background 2.png"), caption="Generated example, 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, regenerating certain objects in the room", use_column_width=True)
-            st.image(Image.open("content/segmentation window.png"), caption="Generated example, regenerating certain objects in the room", use_column_width=True)
-            
+        make_image_row(Image.open("content/output_1.png"), Image.open("content/regen_example.png"))
+        make_image_row(Image.open("content/keep background 2.png"), Image.open("content/output_0.png"))
+        make_image_row(Image.open("content/segmentation window.png"), Image.open("content/output_3.png"))
+
     else:
         make_prompt_row()