TaahKay commited on
Commit
3b0f62c
·
verified ·
1 Parent(s): 12cbf91

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -6,6 +6,7 @@ import torch
6
  import torchaudio
7
  import sys
8
  import os
 
9
 
10
  from utils.audio_processing import trim_audio, download_audio_as_wav
11
  from utils.video_processing import trim_video
@@ -70,11 +71,11 @@ if st.session_state.audio_path and not st.session_state.transcription:
70
  if st.button("Extract Audio"):
71
  st.session_state.audio_ready = True
72
  st.audio(st.session_state.audio_path, format='audio/wav')
73
-
74
  mem = psutil.virtual_memory()
75
  st.write(f"Memory used: {mem.percent}%")
76
  #Detect Language AND FILTER OUT NON-ENGLISH AUDIOS FOR ANALYSIS
77
- segments, info = st.session_state.whisper.transcribe(st.session_state.audio_path, beam_size=5)
78
 
79
  # Convert segments (generator) to full transcription string
80
  st.session_state.transcription = " ".join([segment.text for segment in segments])
 
6
  import torchaudio
7
  import sys
8
  import os
9
+ import time
10
 
11
  from utils.audio_processing import trim_audio, download_audio_as_wav
12
  from utils.video_processing import trim_video
 
71
  if st.button("Extract Audio"):
72
  st.session_state.audio_ready = True
73
  st.audio(st.session_state.audio_path, format='audio/wav')
74
+ time.sleep(5)
75
  mem = psutil.virtual_memory()
76
  st.write(f"Memory used: {mem.percent}%")
77
  #Detect Language AND FILTER OUT NON-ENGLISH AUDIOS FOR ANALYSIS
78
+ segments, info = st.session_state.whisper.transcribe(st.session_state.audio_path, beam_size=1)
79
 
80
  # Convert segments (generator) to full transcription string
81
  st.session_state.transcription = " ".join([segment.text for segment in segments])