File size: 715 Bytes
5353e72
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
0a78294
5353e72
0a78294
 
 
5353e72
0a78294
5353e72
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
#!/usr/bin/python3
# -*- coding: utf-8 -*-
import argparse

from gradio_client import Client, file

from project_settings import project_path


def get_args():
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--filename",
        default=(project_path / "data/test_wavs/paraformer-zh/si_chuan_hua.wav").as_posix(),
        type=str
    )
    args = parser.parse_args()
    return args


def main():
    args = get_args()

    filename = args.filename

    client = Client("hf-audio/whisper-large-v3")
    result = client.predict(
        inputs=file(filename),
        task="transcribe",
        api_name="/predict"
    )
    print(result)
    return


if __name__ == '__main__':
    main()