Update app.py

app.py

@@ -25,7 +25,7 @@ if mode == "Predict Mode":
     st.header("Upload a waste image to find its category")
 
     # Note
-    st.markdown("* Please note that our dataset is trained primarily with images that contain a white background.  Therefore, images with white background would produce maximum accuracy *")
+    st.markdown("* Please note that our dataset is trained primarily with images that contain a white background. Therefore, images with a white background would produce maximum accuracy *")
 
     # Image upload section
     opt = st.selectbox("How do you want to upload the image for classification?", ("Please Select", "Upload image from device"))
@@ -38,7 +38,7 @@ if mode == "Predict Mode":
             try:
                 image = preprocess_image(file)
             except Exception as e:
-                st.error(f"An error occurred: {e}.  Please contact us EcoClim Solutions at EcoClimSolutions.wordpress.com.")
+                st.error(f"An error occurred: {e}. Please contact us EcoClim Solutions at EcoClimSolutions.wordpress.com.")
     try:
         if image is not None:
             st.image(image, width=256, caption='Uploaded Image')
@@ -53,14 +53,10 @@ if mode == "Predict Mode":
                     # Allow user to provide correct label
                     user_label = st.text_input("Enter the correct label:")
                     if user_label:
-                        # Update the model with the user-provided image and label
-                        image = preprocess_image(file)  # preprocess the image again
-                        target = np.zeros(len(labels))
-                        target[labels.index(user_label)] = 1
-                        model.train_on_batch(image[np.newaxis, ...], target)
-                        st.success(f'Thank you for providing feedback. Model has been updated with the new label.')
+                        st.success(f'Thank you for providing feedback. Please switch to "Train Mode" to update the model.')
+
     except Exception as e:
-        st.error(f"An error occurred: {e}.  Please contact us EcoClim Solutions at EcoClimSolutions.wordpress.com.")
+        st.error(f"An error occurred: {e}. Please contact us EcoClim Solutions at EcoClimSolutions.wordpress.com.")
 
 elif mode == "Train Mode":
     # Train the model with a new image and label
@@ -84,4 +80,4 @@ elif mode == "Train Mode":
                 model.train_on_batch(image[np.newaxis, ...], target)
                 st.success(f'Model has been trained with the new image and label.')
         except Exception as e:
-            st.error(f"An error occurred: {e}.  Please contact us EcoClim Solutions at EcoClimSolutions.wordpress.com.")
+            st.error(f"An error occurred: {e}. Please contact us EcoClim Solutions at EcoClimSolutions.wordpress.com.")