gradio/demo/stt_or_tts/run.py
Freddy Boulton 38b16beb16
Switch gpt-j-6b model for gpt2-xl (#4750)
* Switch model and fix demo

* fix remaining demos

---------

Co-authored-by: Abubakar Abid <abubakar@huggingface.co>
2023-06-30 16:30:04 -05:00

26 lines
561 B
Python

import gradio as gr
tts_examples = [
"I love learning machine learning",
"How do you do?",
]
tts_demo = gr.load(
"huggingface/facebook/fastspeech2-en-ljspeech",
title=None,
examples=tts_examples,
description="Give me something to say!",
)
stt_demo = gr.load(
"huggingface/facebook/wav2vec2-base-960h",
title=None,
inputs="mic",
description="Let me try to guess what you're saying!",
)
demo = gr.TabbedInterface([tts_demo, stt_demo], ["Text-to-speech", "Speech-to-text"])
if __name__ == "__main__":
demo.launch()