Upload /content/config.json with huggingface_hub

content/config.json

@@ -0,0 +1,32 @@
+{
+  "architecture": "resnet18",
+  "framework": "pytorch",
+  "task": "image-classification",
+  "num_classes": 10,
+  "classifier": {
+    "type": "Sequential",
+    "layers": [
+      {
+        "Linear": [
+          512,
+          120
+        ]
+      },
+      "ReLU",
+      {
+        "Dropout": 0.2
+      },
+      {
+        "Linear": [
+          120,
+          10
+        ]
+      }
+    ]
+  },
+  "input_size": [
+    3,
+    224,
+    224
+  ]
+}