Update app.py

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 import os
 import subprocess
 from pytube import YouTube
+import subprocess
  
 pipe = pipeline(model="tilos/whisper-small-zh-HK")  # change to "your-username/the-name-you-picked"
 
@@ -17,6 +18,9 @@ def transcribe(audio):
     text = pipe(audio)["text"]
     return text
 
+def gen_sub(video):
+    result = subprocess.check_output(['auto_subtitle',video,'--model', 'medium'])
+    return text
 
 def get_text(url):
     result = pipe(get_audio(url))
@@ -37,6 +41,7 @@ def offline_video(video):
     text = transcribe(audio_file)
     return text
 
+def 
 
 with gr.Blocks() as demo:
 
@@ -51,6 +56,15 @@ with gr.Blocks() as demo:
             allow_flagging="never",
         )
 
+    # audio file input, generate subtitled video
+    gr.Interface(
+            description= "Generate Cantonese subtitled video from video file",
+            fn=gen_sub,
+            inputs="video",
+            outputs="video",
+            allow_flagging="never",
+        )
+            
     # audio file input
     with gr.Row():
         with gr.Column():