ArzEn-LLM
Collection
Models trained on AraZn Dataset • 16 items • Updated • 11
How to use ahmedheakl/arazn-whisper-small with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("automatic-speech-recognition", model="ahmedheakl/arazn-whisper-small") # pip install -U transformers accelerate
# Load model directly
from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq
processor = AutoProcessor.from_pretrained("ahmedheakl/arazn-whisper-small")
model = AutoModelForSpeechSeq2Seq.from_pretrained("ahmedheakl/arazn-whisper-small", device_map="auto")Please see paper & code for more information:
BibTeX:
@article{heakl2024arzen,
title={ArzEn-LLM: Code-Switched Egyptian Arabic-English Translation and Speech Recognition Using LLMs},
author={Heakl, Ahmed and Zaghloul, Youssef and Ali, Mennatullah and Hossam, Rania and Gomaa, Walid},
journal={arXiv preprint arXiv:2406.18120},
year={2024}
}
This model is a fine-tuned version of openai/whisper-small on the None dataset.
The following hyperparameters were used during training:
Base model
openai/whisper-small