Spaces:
Build error
Build error
pustozerov
commited on
Commit
·
e8d85a3
1
Parent(s):
5c2657f
Updated paths for user data.
Browse files
app.py
CHANGED
@@ -11,7 +11,8 @@ from scipy.io.wavfile import write
|
|
11 |
from modules.diarization.nemo_diarization import diarization
|
12 |
|
13 |
FOLDER_WAV_DB = "data/database/"
|
14 |
-
|
|
|
15 |
SAMPLE_RATE = 16000
|
16 |
dataset = load_dataset("pustozerov/crema_d_diarization", split='validation')
|
17 |
|
@@ -47,16 +48,15 @@ if st.button('Try a random sample from the database'):
|
|
47 |
uploaded_file = st.file_uploader("Choose your recording with a speech",
|
48 |
accept_multiple_files=False, type=["mp3", "wav", "ogg"])
|
49 |
if uploaded_file is not None:
|
50 |
-
os.makedirs(
|
51 |
-
for f in glob.glob(FOLDER_WAV_USER + '*'):
|
52 |
-
os.remove(f)
|
53 |
-
save_path = FOLDER_WAV_USER + uploaded_file.name
|
54 |
if ".mp3" in uploaded_file:
|
55 |
sound = AudioSegment.from_mp3(uploaded_file)
|
56 |
elif ".ogg" in uploaded_file:
|
57 |
sound = AudioSegment.from_ogg(uploaded_file)
|
58 |
else:
|
59 |
sound = AudioSegment.from_wav(uploaded_file)
|
|
|
|
|
60 |
sound.export(save_path, format="wav", parameters=["-ac", "1"])
|
61 |
file_name = os.path.basename(save_path).split(".")[0]
|
62 |
audio_file = open(save_path, 'rb')
|
|
|
11 |
from modules.diarization.nemo_diarization import diarization
|
12 |
|
13 |
FOLDER_WAV_DB = "data/database/"
|
14 |
+
FOLDER_USER_DATA = "data/user_data/"
|
15 |
+
FOLDER_USER_DATA_WAV = "data/user_data_wav/"
|
16 |
SAMPLE_RATE = 16000
|
17 |
dataset = load_dataset("pustozerov/crema_d_diarization", split='validation')
|
18 |
|
|
|
48 |
uploaded_file = st.file_uploader("Choose your recording with a speech",
|
49 |
accept_multiple_files=False, type=["mp3", "wav", "ogg"])
|
50 |
if uploaded_file is not None:
|
51 |
+
os.makedirs(FOLDER_USER_DATA, exist_ok=True)
|
|
|
|
|
|
|
52 |
if ".mp3" in uploaded_file:
|
53 |
sound = AudioSegment.from_mp3(uploaded_file)
|
54 |
elif ".ogg" in uploaded_file:
|
55 |
sound = AudioSegment.from_ogg(uploaded_file)
|
56 |
else:
|
57 |
sound = AudioSegment.from_wav(uploaded_file)
|
58 |
+
save_path = FOLDER_USER_DATA_WAV + uploaded_file.name
|
59 |
+
os.makedirs(FOLDER_USER_DATA_WAV, exist_ok=True)
|
60 |
sound.export(save_path, format="wav", parameters=["-ac", "1"])
|
61 |
file_name = os.path.basename(save_path).split(".")[0]
|
62 |
audio_file = open(save_path, 'rb')
|