Update trainning.py

trainning.py

@@ -59,7 +59,7 @@ datagen = ImageDataGenerator(
 
 # Calculate class weights to handle data imbalance
 # class_weights = compute_class_weight('balanced', np.unique(train_labels), train_labels)
-class_weights = compute_class_weight('balanced', np.unique(train_labels), train_labels.flatten())
+class_weights = compute_class_weight(class_weight='balanced',classes=np.unique(train_labels),y= train_labels)
 
 # Model Architecture
 model = tf.keras.models.Sequential([