models/multimodal/config.json

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