Safetensors
wolfgangblack's picture
Upload config.json
593bd85 verified
raw
history blame
1.09 kB
{
"class_names": [
"PG",
"PG13",
"R",
"X",
"XXX"
],
"id2label": {
"0": "PG",
"1": "PG13",
"2": "R",
"3": "X",
"4": "XXX"
},
"image_models": [
"baseresNet18",
"bestresNet50",
"vitRater"
],
"label2id": {
"PG": 0,
"PG13": 1,
"R": 2,
"X": 3,
"XXX": 4
},
"model_type": "multimodal",
"models": {
"baseresNet18": "./models/baseresNet18",
"promptMovieBert": "./models/promptMovieBert",
"promptMovieRoberta": "./models/promptMovieRoberta",
"bestresNet50": "./models/bestresNet50",
"vitRater": "./models/vitRater"
},
"nlp_transformers_model_paths": {
"promptMovieBert": "./models/promptMovieBert",
"promptMovieRoberta": "./models/promptMovieRoberta"
},
"resnet_model_paths": {
"baseresNet18": "./models/baseresNet18",
"bestresNet50": "./models/bestresNet50"
},
"text_models": [
"promptMovieBert",
"promptMovieRoberta"
],
"transformers_version": "4.42.4",
"vit_model_paths": {
"vitRater": "./models/vitRater"
},
"yolo_model_paths": {}
}