Update app.py

app.py

@@ -82,8 +82,12 @@ elif mode == "Train Mode":
                 image = tf.image.resize(image, (224, 224))  # adjust the size as needed
                 image = tf.expand_dims(image, axis=0)
 
+                # Create a dataset with the new image and label
+                dataset = tf.data.Dataset.from_tensor_slices((image, target))
+                dataset = dataset.batch(1)  # Batch size 1 as we are using a single image
+
                 # Train the model
-                model.fit(image, target, epochs=1)  # You might want to adjust the number of epochs
+                model.fit(dataset, 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: