Update app.py

app.py

@@ -2,7 +2,6 @@ import moviepy.editor as mp
 import librosa
 import numpy as np
 import gradio as gr
-from pytube import YouTube
 import subprocess
 
 
@@ -23,11 +22,8 @@ def buffer_n_merge(intervals, buffer=0.1):
     return new_intervals
 
 
-def download_and_process_video(youtube_url, threshold_db, buffer_sec):
-    # Download the YouTube video
-    youtube = YouTube(youtube_url)
-    vidpath = 'downloaded_video'
-    youtube.streams.first().download(filename=vidpath)
+def download_and_process_video(threshold_db, buffer_sec):
+    vidpath = 'uploaded_video.mp4'
 
     # load the video
     video = mp.VideoFileClip(vidpath)
@@ -56,12 +52,12 @@ def download_and_process_video(youtube_url, threshold_db, buffer_sec):
 iface = gr.Interface(
     fn=download_and_process_video,
     inputs=[
-        gr.inputs.Textbox(label="YouTube URL"),
+        gr.inputs.File(label="Video File", type="mp4"),
         gr.inputs.Slider(minimum=1, maximum=70, step=1, default=30, label="Threshold (db)"),
         gr.inputs.Slider(minimum=0, maximum=2, step=0.01, default=0.1, label="Buffer (sec)"),
     ],
     outputs=gr.outputs.Video(label="Processed Video"),
-    title="YouTube Video Silence Remover"
+    title="Video Silence Remover"
 )
 
 iface.launch()