Spaces:
Sleeping
Sleeping
File size: 1,530 Bytes
1be195c cbc9eca 64fc2b0 dad8aee 64fc2b0 1be195c c62444a dad8aee 7c61131 dad8aee 64fc2b0 dad8aee 64fc2b0 dad8aee 64fc2b0 7c61131 64fc2b0 dad8aee 64fc2b0 7c61131 7fddde7 64fc2b0 dad8aee 64fc2b0 6652d81 c62444a 7c61131 64fc2b0 |
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 39 40 41 42 43 44 45 46 47 48 49 |
import torch
import gradio as gr
from transformers import pipeline
MODEL_NAME_V1 = "rngzhi/cs3264-project"
MODEL_NAME_V2 = "rngzhi/cs3264-project-v2"
BATCH_SIZE = 8
FILE_LIMIT_MB = 1000
device = 0 if torch.cuda.is_available() else "cpu"
def load_model(model_version):
model_name = MODEL_NAME_V1 if model_version == 'Model-v1' else MODEL_NAME_V2
return pipeline(
task="automatic-speech-recognition",
model=model_name,
chunk_length_s=30,
device=device,
)
def transcribe(model_version, inputs, task):
if inputs is None:
raise gr.Error("No audio file submitted! Please upload or record an audio file before submitting your request.")
pipe = load_model(model_version)
text = pipe(inputs, batch_size=BATCH_SIZE, generate_kwargs={"task": task}, return_timestamps=True)["text"]
return text
demo = gr.Blocks()
mic_transcribe = gr.Interface(
fn=transcribe,
inputs=[gr.Dropdown(choices=['Model-v1', 'Model-v2'], label="Choose Model Version"), gr.Audio(sources="microphone", type="filepath")],
outputs="text",
)
file_transcribe = gr.Interface(
fn=transcribe,
inputs=[gr.Dropdown(choices=['Model-v1', 'Model-v2'], label="Choose Model Version"), gr.Audio(sources="upload", type="filepath")],
outputs="text",
examples=[["Model-v2", "samples/sample1.WAV", "upload"], ["Model-v2", "samples/sample2.WAV", "upload"]]
)
with demo:
gr.TabbedInterface([file_transcribe, mic_transcribe], ["Audio file", "Microphone"])
demo.launch(debug=True)
|