Martijn Bartelds commited on
Commit
8b8c6ec
1 Parent(s): b587f3a

Update log errors

Browse files
Files changed (1) hide show
  1. neural_acoustic_distance.py +10 -8
neural_acoustic_distance.py CHANGED
@@ -12,6 +12,8 @@ from scipy import signal
12
  from transformers import AutoConfig
13
  from transformers.models.wav2vec2 import Wav2Vec2Model
14
 
 
 
15
 
16
  def play_audio(filename):
17
  audio_file = open(filename, "rb")
@@ -89,15 +91,15 @@ def run(model_id, layer, filename_x, filename_y):
89
  with st.spinner("Measuring distance..."):
90
  feats_x = featurizer(filename_x)
91
  feats_y = featurizer(filename_y)
92
- print('3. Features computed') # test
93
  gcm = aligner(feats_x, feats_y)
94
- print('4. Alignments computed') # test
95
 
96
  d = gcm.normalizedDistance
97
  print("Distance:", d)
98
 
99
  c, n = compute_costs(gcm)
100
- print('5. Costs computed') # test
101
  return d, c, n
102
 
103
 
@@ -132,7 +134,7 @@ except OSError:
132
  )
133
  layer = None
134
 
135
- print('1. Model selected') # test
136
 
137
  st.subheader("Audio file selection:")
138
 
@@ -161,7 +163,7 @@ if filename_x == "custom upload":
161
  if filename_y == "custom upload":
162
  filename_y = st.file_uploader("Choose a file (y-axis)", key="f_y")
163
 
164
- print('2. Files selected') # test
165
 
166
  if filename_x is not None and filename_y is not None and layer is not None:
167
  print(f"\nX: {filename_x}\nY: {filename_y}")
@@ -171,7 +173,7 @@ if filename_x is not None and filename_y is not None and layer is not None:
171
 
172
  fig, axes = plt.subplots(figsize=(4, 2.5))
173
 
174
- print('6. Plot init') # test
175
 
176
  window_size = 9
177
  rate = 20
@@ -199,7 +201,7 @@ if filename_x is not None and filename_y is not None and layer is not None:
199
  plt.savefig("./output/plot.pdf")
200
  st.pyplot(fig)
201
 
202
- print('7. Plot filled') # test
203
 
204
  if os.path.isfile("./output/plot.pdf"):
205
  st.caption(" Visualization of neural acoustic distances\
@@ -215,4 +217,4 @@ if filename_x is not None and filename_y is not None and layer is not None:
215
  with open("./output/plot.pdf", "rb") as file:
216
  btn = st.download_button(label="Download plot", data=file, file_name="plot.pdf", mime="image/pdf")
217
 
218
- print('8. End') # test
 
12
  from transformers import AutoConfig
13
  from transformers.models.wav2vec2 import Wav2Vec2Model
14
 
15
+ from datetime import datetime # test
16
+
17
 
18
  def play_audio(filename):
19
  audio_file = open(filename, "rb")
 
91
  with st.spinner("Measuring distance..."):
92
  feats_x = featurizer(filename_x)
93
  feats_y = featurizer(filename_y)
94
+ print('3. Features computed', datetime.now().strftime('%d-%m-%Y %H:%M:%S')) # test
95
  gcm = aligner(feats_x, feats_y)
96
+ print('4. Alignments computed', datetime.now().strftime('%d-%m-%Y %H:%M:%S')) # test
97
 
98
  d = gcm.normalizedDistance
99
  print("Distance:", d)
100
 
101
  c, n = compute_costs(gcm)
102
+ print('5. Costs computed', datetime.now().strftime('%d-%m-%Y %H:%M:%S')) # test
103
  return d, c, n
104
 
105
 
 
134
  )
135
  layer = None
136
 
137
+ print('1. Model selected', datetime.now().strftime('%d-%m-%Y %H:%M:%S')) # test
138
 
139
  st.subheader("Audio file selection:")
140
 
 
163
  if filename_y == "custom upload":
164
  filename_y = st.file_uploader("Choose a file (y-axis)", key="f_y")
165
 
166
+ print('2. Files selected', datetime.now().strftime('%d-%m-%Y %H:%M:%S')) # test
167
 
168
  if filename_x is not None and filename_y is not None and layer is not None:
169
  print(f"\nX: {filename_x}\nY: {filename_y}")
 
173
 
174
  fig, axes = plt.subplots(figsize=(4, 2.5))
175
 
176
+ print('6. Plot init', datetime.now().strftime('%d-%m-%Y %H:%M:%S')) # test
177
 
178
  window_size = 9
179
  rate = 20
 
201
  plt.savefig("./output/plot.pdf")
202
  st.pyplot(fig)
203
 
204
+ print('7. Plot filled', datetime.now().strftime('%d-%m-%Y %H:%M:%S')) # test
205
 
206
  if os.path.isfile("./output/plot.pdf"):
207
  st.caption(" Visualization of neural acoustic distances\
 
217
  with open("./output/plot.pdf", "rb") as file:
218
  btn = st.download_button(label="Download plot", data=file, file_name="plot.pdf", mime="image/pdf")
219
 
220
+ print('8. End', datetime.now().strftime('%d-%m-%Y %H:%M:%S')) # test