voice-detection-api / check_models.py
aadhi97x's picture
clean commit
6c1314b
raw
history blame contribute delete
565 Bytes
from transformers import AutoConfig
candidates = [
"milsun/wav2vec2-large-xlsr-53-fake-voice-detection",
"Gustking/wav2vec2-large-xlsr-deepfake-audio-classification",
"kgour/wav2vec2-large-xlsr-53-deepfake-detection", # Another common one
"padmalcom/wav2vec2-large-fake-audio-detector"
]
for model in candidates:
try:
print(f"Checking {model}...")
config = AutoConfig.from_pretrained(model)
print(f"SUCCESS: {model} exists. Labels: {config.id2label}")
except Exception as e:
print(f"FAILED: {model} - {e}")