Use gradio 3.11 and remove time constrain

README.md

@@ -4,7 +4,7 @@ emoji: 🐨
 colorFrom: blue
 colorTo: red
 sdk: gradio
-sdk_version: 3.8.2
+sdk_version: 3.11.0
 python_version: 3.7
 app_file: app.py
 pinned: false

app.py

@@ -40,7 +40,7 @@ def change_model(model):
     current_model = model
 
 # Credits https://huggingface.co/spaces/rajesh1729/youtube-video-transcription-with-whisper
-def get_audio(url, start_second):
+def get_audio(url):
     yt = YouTube(url)
     video = yt.streams.filter(only_audio=True).first()
     out_file = video.download(output_path=".")
@@ -48,19 +48,12 @@ def get_audio(url, start_second):
     new_file = base + ".wav"
     os.rename(out_file, new_file)
     a = new_file
-    wav_to_cut = AudioSegment.from_file(a)
-    # pydub does things in milliseconds
-    ten_seconds = 10 * 1000
-    start_second = start_second * 1000
-    first_10_seconds = wav_to_cut[start_second:start_second+ten_seconds]
-    os.remove(new_file)
-    first_10_seconds.export("final_audio.wav", format="wav")
-    return "final_audio.wav"
+    return a
 
 # Credits https://huggingface.co/spaces/jeffistyping/Youtube-Whisperer
-def populate_metadata(link, start_second):
+def populate_metadata(link):
     yt = YouTube(link)
-    audio = get_audio(link, start_second)
+    audio = get_audio(link)
     return yt.thumbnail_url, yt.title, audio, audio
 
 def inference(yt_audio_path):
@@ -95,7 +88,7 @@ demo = gr.Blocks()
 
 # Use your Block object as a context
 with demo:
-    gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>" 
+    gr.Markdown("<h1 style='text-align: center'>" 
                 + title 
                 + "</h1>")
     gr.Markdown(description)
@@ -114,9 +107,6 @@ with demo:
 
         with gr.Row():
             link = gr.Textbox(label="YouTube Link")
-            start_second = gr.Number(label="Select starting point (in seconds) for the transcription",
-                                    value=0,
-                                    precision=0)
         with gr.Row():
             preview_btn = gr.Button("Preview")
         with gr.Row().style(mobile_collapse=False, equal_height=True):
@@ -127,7 +117,7 @@ with demo:
             yt_audio_path = gr.Textbox(visible=False)
 
         preview_btn.click(fn=populate_metadata,
-                        inputs=[link, start_second],
+                        inputs=[link],
                         outputs=[img, title, yt_audio, yt_audio_path])
         
         with gr.Row():