update cahnges

app.py

@@ -15,7 +15,7 @@ from torchvision import transforms
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 WEIGHTS_PATH = "mobilenetv3_gender_weights.pth"
-METADATA_PATH = "metadata1.json"
+METADATA_PATH = "metadata.json"
 
 # Load metadata
 with open(METADATA_PATH, "r") as f:

metadata.json

@@ -1,3 +1,38 @@
 {
-  "sdk": "docker"
-}
+    "sdk": "docker",
+    "model_name": "mobilenetv3_large_100",
+    "num_classes": 2,
+    "class_names": [
+        "Female",
+        "Male"
+    ],
+    "accuracy": 0.9285714285714286,
+    "confusion_matrix": [
+        [
+            104,
+            7
+        ],
+        [
+            9,
+            104
+        ]
+    ],
+    "epochs_trained": 10,
+    "loss_history": [
+        0.6070853605352599,
+        0.4011639428549799,
+        0.3184526763085661,
+        0.27616333987178476,
+        0.24627566992722708,
+        0.21110220202084246,
+        0.13941147615169658,
+        0.1258032820348082,
+        0.13784673257634558,
+        0.12205909436632847
+    ],
+    "classifier_config": {
+        "in_features": 1280,
+        "hidden_dim": 128,
+        "dropout": 0.4
+    }
+}