Update app.py

app.py

@@ -13,6 +13,9 @@ import requests
 nltk.download('punkt_tab')
 import io
 import ffmpeg
+import subprocess
+import tempfile
+import os
 
 model = whisper.load_model("base")
 
@@ -38,6 +41,8 @@ def extract_yt_audio(it, video_url, video_file):
       wav_file_path = temp_file_path.replace(".mp4", ".wav")
       subprocess.run(["ffmpeg", "-i", temp_file_path, "-vn", "-acodec", "pcm_s16le", "-ar", "44100", "-ac", "2", wav_file_path])
       sample = AudioSegment.from_wav(wav_file_path)
+      os.remove(temp_file_path)
+      os.remove(wav_file_path)
     elif it == 'URL':
       sample = AudioSegment.from_file(video_url)  
     else: