Log errors

neural_acoustic_distance.py

@@ -89,12 +89,15 @@ def run(model_id, layer, filename_x, filename_y):
     with st.spinner("Measuring distance..."):
         feats_x = featurizer(filename_x)
         feats_y = featurizer(filename_y)
+        print('3. Features computed') # test
         gcm = aligner(feats_x, feats_y)
+        print('4. Alignments computed') # test
 
         d = gcm.normalizedDistance
         print("Distance:", d)
 
         c, n = compute_costs(gcm)
+        print('5. Costs computed') # test
     return d, c, n
 
 
@@ -129,6 +132,8 @@ except OSError:
     )
     layer = None
 
+print('1. Model selected') # test
+
 st.subheader("Audio file selection:")
 
 filename_x = st.selectbox("Filename (x-axis):",
@@ -156,6 +161,8 @@ if filename_x == "custom upload":
 if filename_y == "custom upload":
     filename_y = st.file_uploader("Choose a file (y-axis)", key="f_y")
 
+print('2. Files selected') # test
+
 if filename_x is not None and filename_y is not None and layer is not None:
     print(f"\nX: {filename_x}\nY: {filename_y}")
 
@@ -164,6 +171,8 @@ if filename_x is not None and filename_y is not None and layer is not None:
 
     fig, axes = plt.subplots(figsize=(4, 2.5))
 
+    print('6. Plot init') # test
+
     window_size = 9
     rate = 20
     x = np.arange(0, len(c) * rate, rate)
@@ -190,6 +199,8 @@ if filename_x is not None and filename_y is not None and layer is not None:
     plt.savefig("./output/plot.pdf")
     st.pyplot(fig)
 
+    print('7. Plot filled') # test
+
     if os.path.isfile("./output/plot.pdf"):
         st.caption(" Visualization of neural acoustic distances\
         per frame (based on wav2vec 2.0) with the pronunciation of\
@@ -203,3 +214,5 @@ if filename_x is not None and filename_y is not None and layer is not None:
 
     with open("./output/plot.pdf", "rb") as file:
         btn = st.download_button(label="Download plot", data=file, file_name="plot.pdf", mime="image/pdf")
+
+    print('8. End') # test