Update README.md

README.md

@@ -3,19 +3,21 @@ language: en
 license: mit
 library_name: tensorflow
 tags:
-  - computer-vision
-  - drowsiness-detection
-  - driver-safety
-  - cnn
-  - tensorflow
+- computer-vision
+- drowsiness-detection
+- driver-safety
+- cnn
+- tensorflow
 model_name: driver-drowsiness-detector
 datasets:
-  - ckcl/drowsiness_dataset
-  - custom
+- ckcl/drowsiness_dataset
+- custom
 metrics:
-  - accuracy
-  - binary-crossentropy
+- accuracy
+- binary-crossentropy
 pipeline_tag: image-classification
+base_model:
+- google/mobilenet_v2_1.0_224
 ---
 
 # Driver Drowsiness Detection Model
@@ -73,4 +75,4 @@ is_drowsy = prediction[0][0] > 0.5
 - Suitable for embedded systems and edge devices
 
 ## License
-This model is released under the MIT License.
+This model is released under the MIT License.