{
  "_name_or_path": "Stanford-TH/GenrePrediction",
  "architectures": [
    "GenreModel"
  ],
  "auto_map": {
    "AutoConfig": "Stanford-TH/GenrePrediction--genre_configuration.GenreConfig",
    "AutoModel": "Stanford-TH/GenrePrediction--genre_model.GenreModel"
  },
  "custom_pipelines": {
    "weighted-genre-classification": {
      "default": {
        "model": {
          "pt": "Stanford-TH/GenrePrediction"
        }
      },
      "impl": "genre_pipe.GenrePredictionPipeline",
      "pt": [
        "AutoModel"
      ],
      "tf": [],
      "type": "text"
    }
  },
  "id2label": {
    "0": "Action",
    "1": "Adventure",
    "10": "Horror",
    "11": "Musical",
    "12": "Mystery",
    "13": "Romance",
    "14": "Sci-Fi",
    "15": "Sport",
    "16": "Thriller",
    "2": "Biography",
    "3": "Comedy",
    "4": "Crime",
    "5": "Documentary",
    "6": "Drama",
    "7": "Family",
    "8": "Fantasy",
    "9": "History"
  },
  "label2id": null,
  "model_type": "custom-bert-base-uncased",
  "torch_dtype": "float32",
  "transformers_version": "4.39.3"
}