Update README.md

README.md

@@ -1,16 +1,18 @@
 ---
 tags:
-  - image-classification
-  - cats-vs-dogs
-  - tensorflow
-  - efficientnet
+- image-classification
+- cats-vs-dogs
+- tensorflow
+- efficientnet
 pipeline_tag: image-classification
 library_name: tensorflow
 datasets:
-  - cats_vs_dogs
-license: mit       # or whichever license you prefer
+- cats_vs_dogs
+license: mit
 metrics:
-  - accuracy
+- accuracy
+base_model:
+- Swardhunter/Efficentnetb07
 ---
 
 # Cats vs Dogs — EfficientNetB0 Classifier
@@ -67,4 +69,4 @@ prob = model.predict(img)[0][0]
 if prob >= 0.5:
     print("Dog 🐶 — confidence:", prob)
 else:
-    print("Cat 🐱 — confidence:", 1 - prob)
+    print("Cat 🐱 — confidence:", 1 - prob)