Update app.py

app.py

@@ -6,8 +6,11 @@ from PIL import Image
 
 
 # Initialization
+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 'upload_choice' not in st.session_state:
-    result = None
     st.session_state['upload_choice'] = 'file_up'
 
 def change_state():
@@ -69,10 +72,12 @@ model_loaded.load_state_dict(torch.load(model_path, map_location='cpu'))
 # Set the model to evaluation mode
 model_loaded.eval()
 
-if img is not None:
-    result = predict_image(image_path=img, model=model_loaded)
+if st.session_state['generate_result'] != 0:
+    if img is not None:
+        result = predict_image(image_path=img, model=model_loaded)
+    st.session_state['generate_result'] = 0
 
-if result is not None:
+if st.session_state['show_result'] != 0:
     if result == 'Bird':
         st.write(":bird:" + " That's a Bird " + ":bird:")
     else: