pustozerov commited on
Commit
e8d85a3
·
1 Parent(s): 5c2657f

Updated paths for user data.

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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
- FOLDER_WAV_USER = "data/user_data/"
 
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(FOLDER_WAV_USER, exist_ok=True)
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')