Update config.json

config.json

@@ -1,5 +1,17 @@
 {
-  "architecture": "ResNet-50",
-  "num_labels": 2,
-  "model_type": "image-classification"
-}
+  "model_name": "smile_classification",
+  "model_type": "pytorch",
+  "models": [
+    {
+      "name": "smile_classification",
+      "path": "./model_round_5.pth",
+      "description": "model weight"
+    },
+
+  ],
+  "default_model": "microsoft/resnet-50",
+  "description": "A smile classification model",
+  "version": "1.0.0",
+  "tags": ["classification"],
+  "license": "apache-2.0"
+}