Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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(
|
| 53 |
pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1")
|
| 54 |
-
pipeline.to(
|
| 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 |
-
|
|
|
|
| 133 |
|
| 134 |
if st.button("Analyze All Audio",key=f"button_all"):
|
| 135 |
if len(valid_files) == 0:
|
| 136 |
-
|
| 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)
|