Gaurav2k
/

IMDB_Recommender

Text Classification

Inference Endpoints

Model card Files Files and versions Community

Gaurav2k commited on Sep 19, 2024

Commit

316462c

·

verified ·

1 Parent(s): 12921e2

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -17,7 +17,7 @@
   "layer_norm_eps": 1e-12,
   "pad_token_id": 0,
   "gradient_checkpointing": false,
-  "num_labels": 2, // Adjust based on your model's labels
   "id2label": {
     "0": "LABEL_0",
     "1": "LABEL_1"
@@ -26,4 +26,4 @@
     "LABEL_0": 0,
     "LABEL_1": 1
   }
-}

   "layer_norm_eps": 1e-12,
   "pad_token_id": 0,
   "gradient_checkpointing": false,
+  "num_labels": 2,
   "id2label": {
     "0": "LABEL_0",
     "1": "LABEL_1"
     "LABEL_0": 0,
     "LABEL_1": 1
   }
+}