Update functions.py

functions.py

@@ -236,7 +236,7 @@ def inference(link, upload, _asr_model):
             audio_file = yt.streams.filter(only_audio=True)[0].download(filename="audio.mp4")
     
             #Get size of audio file
-            audio_size = round(os.path.getsize(path)/(1024*1024),1)
+            audio_size = round(os.path.getsize(audio_file)/(1024*1024),1)
     
             #Check if file is > 24mb, if not then use Whisper API
             if audio_size <= 25:
@@ -249,7 +249,7 @@ def inference(link, upload, _asr_model):
     
                 st.write('File size larger than 24mb, applying chunking and transcription')
     
-                song = AudioSegment.from_file("audio.mp4", format='mp4')
+                song = AudioSegment.from_file(audio_file, format='mp4')
     
                 # PyDub handles time in milliseconds
                 twenty_minutes = 20 * 60 * 1000