Update app file

app.py

@@ -33,9 +33,9 @@ with col1:
         )   
 
 st.write('##')
-    
-col1, col2, col3, col4 = st.columns([1, 2, 2, 1], gap='large')
-with st.container(border = True):
+
+with st.container(border = True):    
+    col1, col2, col3, col4 = st.columns([1, 2, 2, 1], gap='large')
     with col2:
         st.markdown('##### Input File')
         # set input image by upload
@@ -53,11 +53,11 @@ with st.container(border = True):
             options = ['factory_pid.png', 'plant_pid.png', 'processing_pid.png'],
             format_func = radio_func,
         )
-with col3:
-    st.markdown('##### Preview')
-    # visualize input image
-    if image_file is not None:
-        image = Image.open(image_file)
-    else:
-        image = sahi.utils.cv.read_image_as_pil(IMAGE_TO_URL[radio])
-    st.image(image, use_column_width = True)
+    with col3:
+        st.markdown('##### Preview')
+        # visualize input image
+        if image_file is not None:
+            image = Image.open(image_file)
+        else:
+            image = sahi.utils.cv.read_image_as_pil(IMAGE_TO_URL[radio])
+        st.image(image, use_column_width = True)