Spaces:
Runtime error
Runtime error
File size: 1,433 Bytes
0899e82 e0c2b71 7dd7859 0899e82 7dd7859 0899e82 7dd7859 0899e82 |
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 |
import gradio as gr
import torch
import whisper
### ————————————————————————————————————————
title="Whisper to Emotion"
### ————————————————————————————————————————
whisper_model = whisper.load_model("small")
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
def translate(audio):
print("""
—
Sending audio to Whisper ...
—
""")
audio = whisper.load_audio(audio)
audio = whisper.pad_or_trim(audio)
mel = whisper.log_mel_spectrogram(audio).to(whisper_model.device)
_, probs = whisper_model.detect_language(mel)
transcript_options = whisper.DecodingOptions(task="transcribe", fp16 = False)
translate_options = whisper.DecodingOptions(task="translate", fp16 = False)
transcription = whisper.decode(whisper_model, mel, transcript_options)
translation = whisper.decode(whisper_model, mel, translate_options)
print("Language Spoken: " + transcription.language)
print("Transcript: " + transcription.text)
print("Translated: " + translation.text)
return transcription.language
record_input = gr.Audio(source="microphone",type="filepath", show_label=False)
iface = gr.Interface(fn=translate, inputs=record_input, outputs="text")
iface.launch() |