czyoung commited on
Commit
5f60195
·
verified ·
1 Parent(s): 9642585

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -49,9 +49,9 @@ except RuntimeError as e:
49
 
50
  # Instantiate and prepare model for training.
51
  dfModel, dfState, _ = init_df(model_base_dir="DeepFilterNet3")
52
- dfModel.to(device)#torch.device("cuda"))
53
  pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1")
54
- pipeline.to(device)#torch.device("cuda"))
55
 
56
  # Store results for viewing and further processing
57
  if 'results' not in st.session_state:
@@ -129,11 +129,12 @@ if uploaded_file_paths is not None:
129
  st.session_state.results.append([])
130
  while (len(st.session_state.summaries) < len(valid_files)):
131
  st.session_state.summaries.append([])
132
- st.info(f'{len(valid_files)} valid files: {[fi.name for fi in valid_files]}')
 
133
 
134
  if st.button("Analyze All Audio",key=f"button_all"):
135
  if len(valid_files) == 0:
136
- tab.error('Upload file(s) first!')
137
  else:
138
  start_time = time.time()
139
  totalFiles = len(valid_files)
 
49
 
50
  # Instantiate and prepare model for training.
51
  dfModel, dfState, _ = init_df(model_base_dir="DeepFilterNet3")
52
+ dfModel.to(torch.device("cuda"))
53
  pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1")
54
+ pipeline.to(torch.device("cuda"))
55
 
56
  # Store results for viewing and further processing
57
  if 'results' not in st.session_state:
 
129
  st.session_state.results.append([])
130
  while (len(st.session_state.summaries) < len(valid_files)):
131
  st.session_state.summaries.append([])
132
+
133
+ st.info(f'{len(valid_files)} valid files: {[fi.name for fi in valid_files]}')
134
 
135
  if st.button("Analyze All Audio",key=f"button_all"):
136
  if len(valid_files) == 0:
137
+ st.error('Upload file(s) first!')
138
  else:
139
  start_time = time.time()
140
  totalFiles = len(valid_files)