eengel7's picture
Update app.py
2e1f2f2
raw
history blame
898 Bytes
from transformers import pipeline
import gradio as gr
import os
import deepl
import openai
TARGET_LANG = "EN-GB"
deepl_key = os.environ.get('DEEPL')
translator = deepl.Translator(deepl_key)
pipe = pipeline(model="torileatherman/train_first_try") # change to "your-username/the-name-you-picked"
def transcribe(audio):
text_sv = pipe(audio)["text"]
print(f"Audio transcribed: {text_sv}")
text_en = translator.translate_text(text_sv, target_lang=TARGET_LANG).text
print(f"Text translated: {text_en}")
return text_sv, text_en
iface = gr.Interface(
fn=transcribe,
inputs=gr.Audio(source="microphone", type="filepath"),
outputs=[gr.Textbox(label="Transcribed text"),
gr.Textbox(label="English translation")],
title="Swedish speech to english text",
description="Transcribing swedish speech to text and translating to english!",
)
iface.launch()