voice-detection-api / inspect_model.py
aadhi97x's picture
clean commit
6c1314b
raw
history blame contribute delete
355 Bytes
from transformers import AutoModelForAudioClassification, AutoConfig
import torch
model_name = "MelodyMachine/Deepfake-audio-detection"
try:
print(f"Loading config for {model_name}...")
config = AutoConfig.from_pretrained(model_name)
print("ID2LABEL Mapping:")
print(config.id2label)
except Exception as e:
print(f"Error: {e}")