Update README.md

README.md

@@ -68,7 +68,7 @@ This model is a custom-designed architecture developed independently. It is not
 ## Usage
 
 ## Create NoDClassificationLayer class
-```
+```python
 import tensorflow as tf
 from tensorflow.keras import layers, Model, initializers, datasets
 from tensorflow.keras.models import load_model
@@ -197,7 +197,7 @@ class NoDClassificationLayer(layers.Layer):
 ```
 ## Create the load balaced layer class
 
-```
+```python
 import tensorflow as tf
 from tensorflow.keras import layers
 
@@ -275,7 +275,7 @@ class BalancedNoDClassificationLayer(layers.Layer):
 
 ## Load and evaluate saved model
 
-```
+```python
 def evaluate_saved_model(model_path):
     print(f"[INFO] Loading model from '{model_path}'...")