Update app.py

app.py

@@ -1,7 +1,27 @@
 import gradio as gr
 
-with gr.Blocks() as demo:
-    with gr.Tab("Vision"):
-        gr.Interface.load("models/google/vit-base-patch16-224", title="Vision Transformer")
-        #btn2 = gr.Button("Button 2")
-demo.launch()
+title = "GPT-J-6B"
+
+tts_examples = [
+    "I love learning machine learning",
+    "How do you do?",
+]
+
+tts_demo = gr.Interface.load(
+    "huggingface/facebook/fastspeech2-en-ljspeech",
+    title=None,
+    examples=tts_examples,
+    description="Give me something to say!",
+)
+
+stt_demo = gr.Interface.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()