Update app.py

app.py

@@ -77,7 +77,7 @@ 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
-                model.train_on_batch(image[np.newaxis, ...], target)
+                tf.train.batch(image, 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.")