add

app.py

@@ -1,13 +1,19 @@
+import requests
 import gradio as gr
 import os
+import urllib.request
 
 from gradio_client import Client
 
-link = os.environ['link']
-client = Client(link)
+hf_link = os.environ['link']
+client = Client(hf_link)
+local_link = os.environ['local_link']
+local_api = os.environ['local_api']
 
+local_code = requests.head(local_link).status_code
+print(local_code)
 
-def s2t(source, mic, fi, lang):
+def s2t_hf(source, mic, fi, lang):
     if source == 'file':
         in_file = fi
     else:
@@ -22,6 +28,25 @@ def s2t(source, mic, fi, lang):
     print(result)
     return result
 
+def s2t_local(source, mic, fi, lang):
+    if source == 'file':
+        in_file = fi
+    else:
+        in_file = mic
+
+    form_data = {"language": lang}
+    files = {'input_file': open(in_file, 'rb')}
+    print(files)
+    
+    response = requests.post(local_api, files=files, data=form_data, timeout=10000)
+
+    return response.json()
+
+def s2t(source, mic, fi, lang):
+    if local_code==200:
+        return s2t_local(source, mic, fi, lang)
+    else:
+        return s2t_hf(source, mic, fi, lang)
 
 def update_audio_ui(audio_source: str, input_audio_mic, input_audio_file):
     mic = audio_source == "microphone"
@@ -34,7 +59,7 @@ with gr.Blocks() as demo:
     with gr.Group():
         with gr.Row() as audio_box:
             with gr.Column():
-                input_lang = gr.Dropdown(['auto','vi', 'ja', 'en-us', 'cn', 'ko'], label='Language?', value='auto', interactive=True)
+                input_lang = gr.Dropdown(['auto','vi', 'ja', 'en', 'cn', 'ko'], label='Language?', value='en', interactive=True)
                 audio_source = gr.Radio(
                     label="Audio source",
                     choices=["file", "microphone"],
@@ -69,6 +94,7 @@ with gr.Blocks() as demo:
         api_name=False,
     )
 
+
     btn.click(fn=s2t, inputs=[audio_source, input_audio_mic, input_audio_file, input_lang], outputs=[js])
 
 if __name__ == "__main__":