voice-detection-api / download_model.py
aadhi97x's picture
clean commit
6c1314b
raw
history blame contribute delete
406 Bytes
from transformers import AutoModelForAudioClassification, AutoFeatureExtractor
import os
model_name = "MelodyMachine/Deepfake-audio-detection-V2"
print(f"Downloading model: {model_name}...")
try:
AutoFeatureExtractor.from_pretrained(model_name)
AutoModelForAudioClassification.from_pretrained(model_name)
print("Download complete!")
except Exception as e:
print(f"Formatting error: {e}")