Update app.py

app.py

@@ -1,33 +1,31 @@
 import torch
 import scipy
 import gradio as gr
-
-from transformers import set_seed, pipeline
-from transformers import VitsTokenizer, VitsModel
-from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import set_seed
 from datasets import load_dataset, Audio
 
-import speech_to_text, text_to_speech, translation
+import  goai_stt, goai_tts, goai_traduction
 
-language_list = ['mos', 'fra', 'eng']
+#language_list = ['mos', 'fra', 'eng']
+device = 0 if torch.cuda.is_available() else "cpu"
 
 demo = gr.Blocks()
 
-mms_stt = gr.Interface(
-    fn=speech_to_text.transcribe,
+goai_stt = gr.Interface(
+    fn = goai_stt.goai_stt,
     inputs=[
         gr.Audio(sources=["microphone", "upload"], type="filepath"),
-        gr.Dropdown(language_list, label="Language")
+        #gr.Dropdown(language_list, label="Language")
     ],
     outputs="text",
     title="Speech-to-text"
 )
 
-mms_tts = gr.Interface(
-    fn=text_to_speech.synthesize_facebook,
+goai_tts = gr.Interface(
+    fn=goai_tts.goai_tts,
     inputs=[
         gr.Text(label="Input text"),
-        gr.Dropdown(language_list, label="Language")
+        #gr.Dropdown(language_list, label="Language")
     ],
     outputs=[
         gr.Audio(label="Generated Audio", type="numpy")
@@ -35,21 +33,22 @@ mms_tts = gr.Interface(
     title="Text-to-speech"
 )
 
-mms_translate = gr.Interface(
-    fn=translation.translation,
+goai_traduction = gr.Interface(
+    fn=goai_traduction.goai_traduction,
     inputs=[
         gr.Textbox(label="Text", placeholder="Yaa sõama"),
         gr.Dropdown(label="Source Language", choices=["eng_Latn", "fra_Latn", "mos_Latn"]),
         gr.Dropdown(label="Target Language", choices=["eng_Latn", "fra_Latn", "mos_Latn"])
     ],
     outputs=["text"],
-    examples=[["Building a translation demo with Gradio is so easy!", "eng_Latn", "mos_Latn"]],
+    examples=[["Yʋʋm a wãn la b kẽesd biig lekolle?", "mos_Latn", "fra_Latn"]],
     title="Translation Demo",
 )
 
+
 with demo:
     gr.TabbedInterface(
-        [mms_translate, mms_tts, mms_stt],
+        [goai_traduction, goai_tts, goai_stt],
         ["Translation", "Text-to-speech", "Speech-to-text"],
     )