File size: 355 Bytes
6c1314b
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
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}")