English asr_wav2vec2_xls_r_300m_bangla_command_generated_data_finetune TFWav2Vec2ForCTC from hrdipto

Description

Pretrained Wav2vec2 model, adapted from Hugging Face and curated to provide scalability and production-readiness using Spark NLP.asr_wav2vec2_xls_r_300m_bangla_command_generated_data_finetune is a English model originally trained by hrdipto.

NOTE: This model only works on a CPU, if you need to use this model on a GPU device please use asr_wav2vec2_xls_r_300m_bangla_command_generated_data_finetune_gpu

Download Copy S3 URICopied!

How to use


audio_assembler = AudioAssembler() \
    .setInputCol("audio_content") \
    .setOutputCol("audio_assembler")

speech_to_text = Wav2Vec2ForCTC \
    .pretrained("asr_wav2vec2_xls_r_300m_bangla_command_generated_data_finetune", "en")\
    .setInputCols("audio_assembler") \
    .setOutputCol("text")

pipeline = Pipeline(stages=[
  audio_assembler,
  speech_to_text,
])

pipelineModel = pipeline.fit(audioDf)

pipelineDF = pipelineModel.transform(audioDf)

Model Information

Model Name: asr_wav2vec2_xls_r_300m_bangla_command_generated_data_finetune
Compatibility: Spark NLP 4.2.0+
License: Open Source
Edition: Official
Input Labels: [audio_assembler]
Output Labels: [text]
Language: en
Size: 1.2 GB