File size: 409 Bytes
33442e6
c8e15e2
 
33442e6
c8e15e2
33442e6
c8e15e2
33442e6
 
 
 
c8e15e2
33442e6
 
 
 
c8e15e2
33442e6
c8e15e2
33442e6
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
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 ✅")