Update app.py

app.py

@@ -4,10 +4,10 @@ from langchain_openai import ChatOpenAI
 from utils import RefineDataSummarizer
 import os
 
-def transcript(file_dir, model_type):
+def transcript(file_dir, language, model_type):
     model_dir = os.path.join('models', model_type)
     model = whisper.load_model(model_dir)
-    result = model.transcribe(file_dir, language='English', task='transcribe')
+    result = model.transcribe(file_dir, language=language, task='transcribe')
 
     lines = [s['text'] for s in result['segments']]
     text = ''
@@ -37,6 +37,8 @@ with gr.Blocks() as demo:
             file_output = gr.File()
             upload_button = gr.UploadButton("Click to Upload a File", file_types=["audio", "video"], file_count="single")
             upload_button.upload(upload_file, upload_button, file_output)
+            language = gr.Dropdown(
+                ["English", "Chinese"], label="Transcript Language", value="English")
             model_type = gr.Dropdown(
                 [
                     "tiny.en.pt", 
@@ -74,6 +76,7 @@ with gr.Blocks() as demo:
         fn=transcript, 
         inputs=[
             file_output,
+            language,
             model_type
         ],
         outputs=[transcript_text]