Update README.md

README.md

@@ -54,3 +54,23 @@ Input (224, 224, 3)
   ↳ Dropout(0.4)
   ↳ Dense(4, activation='softmax')
 ---
+🚀 How to Load and Use
+
+import tensorflow as tf
+from tensorflow.keras.applications import EfficientNetB0
+from tensorflow.keras.layers import Dense, GlobalAveragePooling2D, Dropout, BatchNormalization
+from tensorflow.keras.models import Model
+
+# Reconstruct Model Architecture
+base_model = EfficientNetB0(weights=None, include_top=False, input_shape=(224, 224, 3))
+x = base_model.output
+x = GlobalAveragePooling2D()(x)
+x = BatchNormalization()(x)
+x = Dense(256, activation='relu')(x)
+x = Dropout(0.4)(x)
+outputs = Dense(4, activation='softmax')(x)
+
+model = Model(inputs=base_model.input, outputs=outputs)
+
+# Load Weights
+model.load_weights("efficientnetb0_finetuned_brain_mri.keras")