xiaoyunchong.xyc commited on
Commit
feb23b0
·
1 Parent(s): 8ced29b

fix: upgrade gradio to 5.23.0 to fix TypeError in api_info

Browse files
Files changed (3) hide show
  1. README.md +1 -1
  2. app_cpu.py +124 -0
  3. requirements.txt +1 -1
README.md CHANGED
@@ -4,7 +4,7 @@ emoji: 🚀
4
  colorFrom: blue
5
  colorTo: indigo
6
  sdk: gradio
7
- sdk_version: 5.9.1
8
  app_file: app.py
9
  pinned: true
10
  license: apache-2.0
 
4
  colorFrom: blue
5
  colorTo: indigo
6
  sdk: gradio
7
+ sdk_version: 5.23.0
8
  app_file: app.py
9
  pinned: true
10
  license: apache-2.0
app_cpu.py ADDED
@@ -0,0 +1,124 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ from huggingface_hub import snapshot_download
3
+
4
+ MODEL_CACHE_DIR = "./models"
5
+ SENSE_VOICE_LOCAL_PATH = os.path.join(MODEL_CACHE_DIR, "SenseVoiceSmall")
6
+ PARAFORMER_LOCAL_PATH = os.path.join(MODEL_CACHE_DIR, "paraformer-zh")
7
+ VAD_LOCAL_PATH = os.path.join(MODEL_CACHE_DIR, "fsmn-vad")
8
+ PUNC_LOCAL_PATH = os.path.join(MODEL_CACHE_DIR, "ct-punc")
9
+
10
+ os.makedirs(MODEL_CACHE_DIR, exist_ok=True)
11
+
12
+
13
+ def download_if_missing(repo_id, local_path, name):
14
+ if not os.path.exists(local_path):
15
+ print(f"Downloading {name}...")
16
+ snapshot_download(repo_id=repo_id, local_dir=local_path, ignore_patterns=["*.onnx"])
17
+ print(f"{name} ready.")
18
+ else:
19
+ print(f"{name} found locally.")
20
+
21
+
22
+ download_if_missing("FunAudioLLM/SenseVoiceSmall", SENSE_VOICE_LOCAL_PATH, "SenseVoice")
23
+ download_if_missing("funasr/paraformer-zh", PARAFORMER_LOCAL_PATH, "Paraformer-zh")
24
+ download_if_missing("funasr/fsmn-vad", VAD_LOCAL_PATH, "FSMN-VAD")
25
+ download_if_missing("funasr/ct-punc", PUNC_LOCAL_PATH, "CT-Punc")
26
+
27
+ import gradio as gr
28
+ import time
29
+ import tempfile
30
+ from funasr import AutoModel
31
+ from funasr.utils.postprocess_utils import rich_transcription_postprocess
32
+
33
+ loaded_models = {}
34
+
35
+
36
+ def get_model(pipeline):
37
+ if pipeline in loaded_models:
38
+ return loaded_models[pipeline]
39
+
40
+ if pipeline == "sensevoice":
41
+ model = AutoModel(
42
+ model=SENSE_VOICE_LOCAL_PATH,
43
+ vad_model=VAD_LOCAL_PATH,
44
+ vad_kwargs={"max_single_segment_time": 30000},
45
+ device="cpu",
46
+ disable_update=True,
47
+ hub="hf",
48
+ )
49
+ elif pipeline == "paraformer":
50
+ model = AutoModel(
51
+ model=PARAFORMER_LOCAL_PATH,
52
+ vad_model=VAD_LOCAL_PATH,
53
+ punc_model=PUNC_LOCAL_PATH,
54
+ device="cpu",
55
+ disable_update=True,
56
+ hub="hf",
57
+ )
58
+ else:
59
+ raise ValueError(f"Unknown pipeline: {pipeline}")
60
+
61
+ loaded_models[pipeline] = model
62
+ return model
63
+
64
+
65
+ def transcribe(audio_input, pipeline_type):
66
+ if audio_input is None:
67
+ return "Please upload or record audio.", ""
68
+
69
+ model = get_model(pipeline_type)
70
+
71
+ t0 = time.time()
72
+ if pipeline_type == "sensevoice":
73
+ res = model.generate(
74
+ input=audio_input, cache={}, language="auto",
75
+ use_itn=True, batch_size_s=60, merge_vad=True, merge_length_s=15,
76
+ )
77
+ else:
78
+ res = model.generate(input=audio_input)
79
+
80
+ text = rich_transcription_postprocess(res[0]["text"])
81
+ elapsed = time.time() - t0
82
+
83
+ metrics = f"Time: {elapsed:.2f}s | Model: {pipeline_type} | Device: CPU"
84
+ return metrics, text
85
+
86
+
87
+ with gr.Blocks(title="FunASR Demo") as demo:
88
+ gr.Markdown("""
89
+ # FunASR: Speech Recognition Demo
90
+
91
+ Industrial-grade ASR toolkit. Upload audio and get transcription instantly.
92
+
93
+ - **SenseVoice**: Multi-task (ASR + emotion + events), 5 languages, ultra-fast
94
+ - **Paraformer**: Non-autoregressive Chinese ASR with punctuation
95
+
96
+ [GitHub](https://github.com/modelscope/FunASR) | [Docs](https://modelscope.github.io/FunASR/) | [pip install funasr](https://pypi.org/project/funasr/)
97
+ """)
98
+
99
+ audio_input = gr.Audio(label="Upload or Record Audio", sources=["upload", "microphone"], type="filepath")
100
+
101
+ pipeline_type = gr.Dropdown(
102
+ choices=["sensevoice", "paraformer"],
103
+ label="Model",
104
+ value="sensevoice"
105
+ )
106
+
107
+ btn = gr.Button("Transcribe", variant="primary")
108
+
109
+ metrics_out = gr.Textbox(label="Metrics", lines=1)
110
+ text_out = gr.Textbox(label="Transcription", lines=8)
111
+
112
+ btn.click(transcribe, inputs=[audio_input, pipeline_type], outputs=[metrics_out, text_out])
113
+
114
+ gr.Markdown("""
115
+ ### Install & Use Locally
116
+ ```python
117
+ pip install funasr
118
+ from funasr import AutoModel
119
+ model = AutoModel(model="funasr/paraformer-zh", hub="hf", vad_model="funasr/fsmn-vad", punc_model="funasr/ct-punc")
120
+ result = model.generate(input="audio.wav")
121
+ ```
122
+ """)
123
+
124
+ demo.queue().launch()
requirements.txt CHANGED
@@ -4,4 +4,4 @@ torchaudio
4
  huggingface_hub
5
  pydub
6
  requests
7
- gradio>=5.0.0,<6.0.0
 
4
  huggingface_hub
5
  pydub
6
  requests
7
+ gradio==5.23.0