config.json

{
  "architectures": ["VoiceShieldForAudioClassification"],
  "auto_map": {
    "AutoConfig": "modeling_voiceshield.VoiceShieldConfig",
    "AutoModelForAudioClassification": "modeling_voiceshield.VoiceShieldForAudioClassification"
  },
  "base_model": "openai/whisper-small",
  "model_type": "voiceshield",
  "num_labels": 2,
  "id2label": {
    "0": "safe",
    "1": "malicious"
  },
  "label2id": {
    "safe": 0,
    "malicious": 1
  },
  "torch_dtype": "float32",
  "transformers_version": "4.36.0"
}