Views
No views yet
1from transformers import WhisperProcessor
2import torch
3from predict import WhisperClassifier, predict, load_model
4
5# Load processor and model
6processor = WhisperProcessor.from_pretrained("ahmad1703/whis_ee")
7model, device = load_model("model.pth")
8
9# Example prediction
10audio_path = "path/to/your/audio_file.wav"
11result = predict(model, audio_path, processor, device)
12print(f"Prediction: {result}") # 0 or 1@misc{whisperclassifier2025,
author = {[Your Name]},
title = {WhisperClassifier for Binary Audio Classification},
year = {2025},
publisher = {HuggingFace},
howpublished = {\url{https://huggingface.co/ahmad1703/whis_ee}}
}