Update app.py

app.py

@@ -7,6 +7,7 @@ from pytube import YouTube
 import os
 import sys
 import subprocess
+import re
 
 loaded_model = whisper.load_model("base")
 current_size = 'base'
@@ -14,17 +15,20 @@ current_size = 'base'
 def download_video(link):
 
     yt = YouTube(link)
+    vid = yt.thumbnail_url.split('vi/')[1].split('/')[0]
+    print(vid)
 
-    return yt.streams.filter(file_extension='mp4')[0].download(filename='video.mp4')
+    return yt.streams.filter(file_extension='mp4')[0].download(filename=f"{vid}.mp4")
 
 def inference(link):
 
   yt = YouTube(link)
-  
-  audio_path = yt.streams.filter(only_audio=True)[0].download(filename="audio.mp4")
+  vid = yt.thumbnail_url.split('vi/')[1].split('/')[0]
+
+  audio_path = yt.streams.filter(only_audio=True)[0].download(filename= f"{vid}.mp3")
   print(f'audio path :  {audio_path}')
   video_path  = download_video(link)
-
+  
   #video_path = yt.streams.filter(file_extension='mp4')[0].download(filename='video.mp4'
   
   options = dict(beam_size=5, best_of=5, fp16 = False)
@@ -86,7 +90,7 @@ with block:
             
             input_video  = gr.Video()
             output_video = gr.Video()
-             
+
           btn = gr.Button("Generate Captions")       
           
           # Events