audio_model / model.py
Simma7's picture
Update model.py
33442e6 verified
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 ✅")