Update app.py
Browse files
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=
|
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])
|