Update config.json

config.json

@@ -4,6 +4,7 @@
   "depth_coefficient": 1.0,
   "width_coefficient": 1.0,
   "dropout_rate": 0.2,
-  "num_classes": 1000,
-  "image_size": 224
+  "num_labels": 1000,
+  "image_size": 224,
+  "framework": "pytorch"
 }