Spaces:
Sleeping
Sleeping
Commit History
-
Update requirements.txt
Verified signature90d99ef
-
Update requirements.txt
Verified signature3248f23 -
Rename requirement.txt to requirements.txt
Verified signature25a7548 -
Update requirement.txt
Verified signature356763f -
Update requirement.txt
Verified signature9a95c8d -
Upload app.py
Verified signaturebbc6a68 -
app.py
Verified signatureimport torch import gradio as gr from transformers import ( AutomaticSpeechRecognitionPipeline, WhisperForConditionalGeneration, WhisperTokenizer, WhisperProcessor, ) from peft import PeftModel, PeftConfig peft_model_id = "Moustapha91/whisper-small-wolof" language = "French" task = "transcribe" peft_config = PeftConfig.from_pretrained(peft_model_id) model = WhisperForConditionalGeneration.from_pretrained( peft_config.base_model_name_or_path, device_map="auto" # On supprime la quantization en 8 bits ) model = PeftModel.from_pretrained(model, peft_model_id) tokenizer = WhisperTokenizer.from_pretrained(peft_config.base_model_name_or_path, language=language, task=task) processor = WhisperProcessor.from_pretrained(peft_config.base_model_name_or_path, language=language, task=task) feature_extractor = processor.feature_extractor forced_decoder_ids = processor.get_decoder_prompt_ids(language=language, task=task) pipe = AutomaticSpeechRecognitionPipeline(model=model…396d112 -
initial commit
Verified signature5855f42