from transformers import AutoModelForAudioClassification, AutoProcessor
import torch

model_name = "Simma7/audio_model"

print("[INFO] Loading model...")

model = AutoModelForAudioClassification.from_pretrained(
    model_name,
    trust_remote_code=True
)

processor = AutoProcessor.from_pretrained(
    model_name,
    trust_remote_code=True
)

model.eval()

print("[SUCCESS] Model loaded successfully ✅")