| { | |
| "_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" | |
| } | |