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 |
|