Upload config.json with huggingface_hub

config.json

@@ -2,7 +2,12 @@
   "architectures": [
     "TinyCNN"
   ],
-  "num_labels": 10,
-  "num_channels": 3,
-  "image_size": 224
+  "input_size": [
+    3,
+    224,
+    224
+  ],
+  "num_classes": 10,
+  "model_type": "torchscript",
+  "library_name": "pytorch"
 }