Update app file

app.py

@@ -20,7 +20,7 @@ st.title('P&ID Object Detection')
 st.subheader(' Identify valves and pumps with deep learning model ', divider='rainbow')
 st.caption('Developed by Deep Drawings Co.')
 
-col1, col2, col3 = st.columns(3, gap='large')
+col1, col2, col3 = st.columns(3, gap='medium')
 with col1:
     with st.expander('How to use it'):
         st.markdown(
@@ -35,23 +35,24 @@ with col1:
 st.write('##')
     
 col1, col2, col3, col4 = st.columns([1, 2, 2, 1], gap='large')
-with col2:
-    st.markdown('##### Input File')
-    # set input image by upload
-    image_file = st.file_uploader("Upload your diagram", type=["pdf"])
-    # set input images from examples
-    def radio_func(option):
-        option_to_id = {
-            'factory_pid.png' : 'A',
-            'plant_pid.png' : 'B',
-            'processing_pid.png' : 'C',
-        }
-        return option_to_id[option]
-    radio = st.radio(
-        'Or select from example diagrams',
-        options = ['factory_pid.png', 'plant_pid.png', 'processing_pid.png'],
-        format_func = radio_func,
-    )
+with st.container(border = True):
+    with col2:
+        st.markdown('##### Input File')
+        # set input image by upload
+        image_file = st.file_uploader("Upload your diagram", type=["pdf"])
+        # set input images from examples
+        def radio_func(option):
+            option_to_id = {
+                'factory_pid.png' : 'A',
+                'plant_pid.png' : 'B',
+                'processing_pid.png' : 'C',
+            }
+            return option_to_id[option]
+        radio = st.radio(
+            'Or select from example diagrams',
+            options = ['factory_pid.png', 'plant_pid.png', 'processing_pid.png'],
+            format_func = radio_func,
+        )
 with col3:
     st.markdown('##### Preview')
     # visualize input image