Update app.py

app.py

@@ -13,6 +13,8 @@ if 'generate_result' not in st.session_state:
     st.session_state['generate_result'] = 0
 if 'show_result' not in st.session_state:
     st.session_state['show_result'] = 0
+if 'number_of_files' not in st.session_state:
+    st.session_state['number_of_files'] = 0
 if 'upload_choice' not in st.session_state:
     st.session_state['upload_choice'] = 'file_up'
 
@@ -34,6 +36,10 @@ st.toggle(label="Webcam",  help="Click on to use webcam, off to upload a file",
 # Use state to know whether to show file_uploader or webcam
 if st.session_state['upload_choice'] == 'file_up':
     img = st.file_uploader(label="Upload a photo of a squirrel or bird", type=['png', 'jpg'])
+    if len(img) > 0:
+        st.session_state['number_of_files'] = 1
+    else:
+        st.session_state['number_of_files'] = 0
 else:
     st.camera_input(label="Webcam")
 
@@ -98,6 +104,11 @@ model_loaded.eval()
 # Toggle view of model output in UI
 ###
 #####
+if st.session_state['upload_choice'] == 'file_up' and st.session_state['number_of_files'] == 1:
+    st.session_state['generate_result'] = 1
+    st.session_state['show_result'] = 1
+    
+
 if st.session_state['generate_result'] != 0:
     if img is not None:
         result = predict_image(image_path=img, model=model_loaded)