Update app.py
Browse files
app.py
CHANGED
@@ -142,7 +142,7 @@ if st.button('Сгенерировать потери'):
|
|
142 |
stoi_mass=[stoi_orig, stoi_lossy, stoi_enhanced]
|
143 |
|
144 |
|
145 |
-
pesq = PESQ(16000, 'nb')
|
146 |
data_clean = data_clean.cpu().numpy()
|
147 |
data_lossy = data_lossy.detach().cpu().numpy()
|
148 |
data_enhanced = data_enhanced.cpu().numpy()
|
@@ -156,7 +156,7 @@ if st.button('Сгенерировать потери'):
|
|
156 |
pesq_lossy = np.array(pesq(torch.tensor(data_lossy), torch.tensor(data_clean)))
|
157 |
pesq_enhanced = np.array(pesq(torch.tensor(data_enhanced), torch.tensor(data_clean)))
|
158 |
|
159 |
-
psq_mas=[pesq_orig, pesq_lossy, pesq_enhanced]
|
160 |
|
161 |
|
162 |
|
@@ -164,7 +164,7 @@ if st.button('Сгенерировать потери'):
|
|
164 |
|
165 |
df['Audio'] = ['Clean', 'Lossy', 'Enhanced']
|
166 |
|
167 |
-
df['PESQ'] = psq_mas
|
168 |
|
169 |
df['STOI'] = stoi_mass
|
170 |
|
|
|
142 |
stoi_mass=[stoi_orig, stoi_lossy, stoi_enhanced]
|
143 |
|
144 |
|
145 |
+
"""pesq = PESQ(16000, 'nb')
|
146 |
data_clean = data_clean.cpu().numpy()
|
147 |
data_lossy = data_lossy.detach().cpu().numpy()
|
148 |
data_enhanced = data_enhanced.cpu().numpy()
|
|
|
156 |
pesq_lossy = np.array(pesq(torch.tensor(data_lossy), torch.tensor(data_clean)))
|
157 |
pesq_enhanced = np.array(pesq(torch.tensor(data_enhanced), torch.tensor(data_clean)))
|
158 |
|
159 |
+
psq_mas=[pesq_orig, pesq_lossy, pesq_enhanced]"""
|
160 |
|
161 |
|
162 |
|
|
|
164 |
|
165 |
df['Audio'] = ['Clean', 'Lossy', 'Enhanced']
|
166 |
|
167 |
+
#df['PESQ'] = psq_mas
|
168 |
|
169 |
df['STOI'] = stoi_mass
|
170 |
|