Update app.py

app.py

@@ -63,7 +63,7 @@ elif mode == "Train Mode":
     st.header("Train the model with a new image and label")
 
     # Image upload section
-    file = st.file_uploader('Select', type=['jpg', 'png', 'jpeg'])
+     file = st.file_uploader('Select', type=['jpg', 'png', 'jpeg'])
     if file:
         try:
             image = preprocess_image(file)
@@ -77,7 +77,14 @@ elif mode == "Train Mode":
                 # Update the model with the user-provided image and label
                 target = np.zeros(len(labels))
                 target[labels.index(user_label)] = 1
-                tf.train.batch(image, target)
+
+                # Reshape the image to match the model's input shape
+                image = tf.image.resize(image, (224, 224))  # adjust the size as needed
+                image = tf.expand_dims(image, axis=0)
+
+                # Train the model
+                model.fit(image, target, epochs=1)  # You might want to adjust the number of epochs
+
                 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.")