Update main.py

main.py

@@ -166,7 +166,7 @@ audio_to_text_interface = gr.Interface(
 )
 
 # Gradio Demo
-demo = gr.TabbedInterface([text_to_audio_interface, audio_to_text_interface], ["Text - Audio", "Audio - Text"])
+#demo = gr.TabbedInterface([text_to_audio_interface, audio_to_text_interface], ["Text - Audio", "Audio - Text"])
 
 # Custom CSS for centered links
 custom_css = """
@@ -201,6 +201,29 @@ Explore the tabs to try the **Text - Audio** and **Audio - Text** modes!
 </div>
 
 """
-
-# Launch Gradio App
-demo.launch(share=True, description=description)
+with gr.Blocks() as demo:
+    gr.Markdown(description)
+    with gr.Tabs():
+        with gr.TabItem("Text - Audio"):
+            text_to_audio_interface = gr.Interface(
+                fn=infer_text_to_audio_gr,
+                inputs=gr.Textbox(label="Input Text"),
+                outputs=gr.Audio(label="Audio Answer"),
+                title="T2S",
+                description="Model in text to audio mode",
+                allow_flagging='never',
+            )
+            text_to_audio_interface.render()
+        with gr.TabItem("Audio - Text"):
+            audio_to_text_interface = gr.Interface(
+                fn=infer_audio_to_text_gr,
+                inputs=gr.Audio(type="filepath", label="Input Audio"),
+                outputs=gr.Textbox(label="Text Answer"),
+                title="S2T",
+                description="Model in audio to text mode",
+                allow_flagging='never'
+            )
+            audio_to_text_interface.render()
+
+# Launch the demo
+demo.launch(share=True)