Update app.py

app.py

@@ -96,12 +96,26 @@ if opt == 'Upload image from device':
     if file:
         image = preprocess_image(file)
 
+# initialize the session state with a default value
+if 'training_mode' not in st.session_state:
+  st.session_state.training_mode = False
+
+# create a button widget
+button = st.button('Toggle training mode')
+
+# update the session state when the button is clicked
+if button:
+  st.session_state.training_mode = not st.session_state.training_mode
+
+# display the current training mode
+st.write(f'Training mode: {st.session_state.training_mode}')
+
 try:
     if image is not None:
         st.image(image, width=256, caption='Uploaded Image')
         if st.button('Predict'):
-            prediction = model.predict(image[np.newaxis, ...])
+            # call the model with the training mode argument
+            prediction = model.predict(image[np.newaxis, ...], training=st.session_state.training_mode)
             st.success(f'Prediction: {labels[np.argmax(prediction[0], axis=-1)]}')
 except Exception as e:
     st.error(f"An error occurred: {e}.  Please contact us EcoClim Solutions at EcoClimSolutions.wordpress.com.")
-