Skip to content

Latest commit

 

History

History

t5_whisper_medical

T5 Whisper Medical Error Correction

Input

TEXT file from whisper medium.

こんにちは、先生。最近手足の経連があります。
こんにちは。手足の経連がある場合、心臓の問題が考えられます。
経連の発作がいつ頻繁に起こるかを教えていただけますか?
経連はほとんど毎日です。

Output

Text after error correction.

こんにちは、先生。最近手足の痙攣があります。こんにちは。手足の痙攣がある場合、心臓の問題が考えられます。痙攣の発作がいつ頻繁に起こるかを教えていただけますか?痙攣はほとんど毎日です。

Usage

Automatically downloads the onnx and prototxt files on the first run. It is necessary to be connected to the Internet while downloading.

For the sample text,

$ python3 t5_whisper_medical.py

If you want to specify the text or pdf file, put the file path after the -i option.

$ python3 t5_whisper_medical.py -i FILE_PATH

Reference

Framework

PyTorch

Model Format

ONNX opset=12

Netron