Update README.md

README.md

@@ -113,12 +113,21 @@ Input (224, 224, 3)
 ---
 🚀 How to Load and Use
 
+import os
+import urllib.request
 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
+weights_path = "efficientnetb0_finetuned_brain_mri.keras"
+model_url = "https://huggingface.co/starpreeda/BrainTumorTest/resolve/main/efficientnetb0_finetuned_brain_mri.keras"
+
+if not os.path.exists(weights_path):
+    print("Downloading model weights...")
+    urllib.request.urlretrieve(model_url, weights_path)
+    print("Download completed!")
+
 base_model = EfficientNetB0(weights=None, include_top=False, input_shape=(224, 224, 3))
 x = base_model.output
 x = GlobalAveragePooling2D()(x)
@@ -126,10 +135,9 @@ 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")
+model.load_weights(weights_path)
+print("Model is ready for use!")
 ```
 📊 Dataset & Training Data