Spaces:
Sleeping
Sleeping
File size: 852 Bytes
93ec611 9f1070e 9f13a48 03b7abb 9f13a48 03b7abb 78a3e26 9f13a48 e669d58 9f13a48 93ec611 9f1070e 93ec611 d818586 19001c8 9f1070e e669d58 e460592 19001c8 9f1070e 93ec611 62e3412 |
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 |
import gradio as gr
import spaces
import torch
import io
import os.path
import urllib.request
import whisper
os.makedirs("/home/user/.cache/whisper")
urllib.request.urlretrieve(
"https://openaipublic.azureedge.net/main/whisper/models/d7440d1dc186f76616474e0ff0b3b6b879abc9d1a4926b7adfa41db2d497ab4f/medium.en.pt",
"/home/user/.cache/whisper/medium.en.pt"
)
writer = whisper.utils.get_writer("srt", "/dev/null")
@spaces.GPU
def generate(file, progress=gr.Progress(track_tqdm=True)):
# get file to type bytes somehow
model = whisper.load_model("medium.en", device="cuda")
audio = whisper.load_audio(file)
result = model.transcribe(audio, verbose=False)
out = io.StringIO()
writer.write_result(result, out)
return out.getvalue()
gr.Interface(fn=generate, inputs=gr.File(type="filepath"), outputs=gr.Text()).launch() |