Updated app.py

app.py

@@ -47,16 +47,13 @@ def main(options):
     
     
     with col1:
-        if 'c01' not in st.session_state:
-            st.session_state.c01=False
-        c01_state=st.session_state.c01
-        c01_state=st.sidebar.checkbox('Take a picture for prediction',value=c01_state)
-
-        if 'c02' not in st.session_state:
-            st.session_state.c02=False
-        c02_state=st.session_state.c02
-        c02_state=st.sidebar.checkbox('Detect',value=c02_state)
-        if c01_state:
+       
+        
+        if st.checkbox('Take a picture for prediction'):
+            if 'c02' not in st.session_state:
+                st.session_state.c02=False
+            c02_state=st.session_state.c02
+            c02_state=st.sidebar.checkbox('Detect',value=c02_state)
            
             with col1:
                 image, original_image= upload()
@@ -65,7 +62,7 @@ def main(options):
                     # Check if original_image is not None
                     pass
                     st.session_state.c02=c02_state
-            st.session_state.c01=c01_state
+            
     with col2:
         st.warning('Wait for few seconds!!')
         progress_bar = st.progress(0.0)