Tech: Change Model type in config to enable inference API (#16)

- tech: update config to enable inference (57f695327786b639286dbcb4844bfab38d3bb111)

config.json

@@ -1,23 +1,35 @@
 {
-  "model_type": "lstm",
-  "base_model": "LSTM",
+  "model_type": "custom",
+  "architectures": ["LSTM"],
   "library_name": "tensorflow",
-  "vocab_size": 500,
-  "embedding_dim": 64,
-  "hidden_size": 64,
-  "num_layers": 2,
-  "dropout_rate": 0.2,
-  "max_sequence_length": 10,
-  "batch_size": 8,
-  "epochs": 50,
-  "loss_function": "sparse_categorical_crossentropy",
-  "optimizer": "adam",
-  "metrics": ["accuracy"],
+  "task_specific_params": {
+    "text-classification": {
+      "vocab_size": 500,
+      "embedding_dim": 64,
+      "hidden_size": 64,
+      "num_layers": 2,
+      "dropout_rate": 0.2,
+      "max_sequence_length": 10
+    }
+  },
+  "training_params": {
+    "batch_size": 8,
+    "epochs": 50,
+    "loss_function": "sparse_categorical_crossentropy",
+    "optimizer": "adam",
+    "metrics": ["accuracy"]
+  },
   "train_data_size": 5000,
-  "categories": [
-    "Lifestyle",
-    "Movies & Entertainment",
-    "Food & Dining",
-    "Others"
-  ]
-}
+  "id2label": {
+    "0": "Lifestyle",
+    "1": "Movies & Entertainment",
+    "2": "Food & Dining",
+    "3": "Others"
+  },
+  "label2id": {
+    "Lifestyle": 0,
+    "Movies & Entertainment": 1,
+    "Food & Dining": 2,
+    "Others": 3
+  }
+}