Teapack1 commited on
Commit
c0fb8f9
1 Parent(s): 4290093

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -13
app.py CHANGED
@@ -2,25 +2,20 @@ import gradio as gr
2
  from transformers import pipeline
3
  import numpy as np
4
 
5
- transcriber = pipeline("automatic-speech-recognition", model="openai/whisper-tiny.en")
6
 
7
- def transcribe(stream, new_chunk):
8
- sr, y = new_chunk
9
  y = y.astype(np.float32)
10
  y /= np.max(np.abs(y))
11
 
12
- if stream is not None:
13
- stream = np.concatenate([stream, y])
14
- else:
15
- stream = y
16
- return stream, transcriber({"sampling_rate": sr, "raw": stream})["text"]
17
 
18
 
19
  demo = gr.Interface(
20
- fn=transcribe,
21
- inputs = gr.Audio(sources=["microphone"], streaming=True),
22
- outputs = gr.outputs.Textbox(),
23
- live=True,
24
  )
25
 
26
- demo.launch(debug=True, share=True)
 
2
  from transformers import pipeline
3
  import numpy as np
4
 
5
+ transcriber = pipeline("automatic-speech-recognition", model="openai/whisper-base.en")
6
 
7
+ def transcribe(audio):
8
+ sr, y = audio
9
  y = y.astype(np.float32)
10
  y /= np.max(np.abs(y))
11
 
12
+ return transcriber({"sampling_rate": sr, "raw": y})["text"]
 
 
 
 
13
 
14
 
15
  demo = gr.Interface(
16
+ transcribe,
17
+ gr.Audio(sources=["microphone"]),
18
+ "text",
 
19
  )
20
 
21
+ demo.launch()