File size: 642 Bytes
8e4d995
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
from transformers import pipeline
import gradio as gr
import time

asr = pipeline("automatic-speech-recognition", "../models/20220513-212939/_lm", device=0)
# asr = pipeline("automatic-speech-recognition", "YSU/aspram")

def transcribe(audio, state=""):
    time.sleep(5)
    text = asr(audio)["text"]
    state += text + " "
    print('chunk!')
    return state, state

gr.Interface(
    fn=transcribe, 
    inputs=[
        gr.inputs.Audio(source="microphone", type="filepath"), 
        "state"
    ],
    outputs=[
        "textbox",
        "state"
    ],
    live=True,
    css='body {background-color: rgba(240, 200, 192)}',
).launch()