Update logging

neural_acoustic_distance.py

@@ -128,7 +128,6 @@ if model_id == "other":
                              key="model")
 
 print(f"\n### Start new run\n") # test
-print('0. RAM memory available:', round(psutil.virtual_memory()[1] / (1024.0 ** 3), 1), 'GB') # test
 
 try:
     cfg = AutoConfig.from_pretrained(model_id)
@@ -224,4 +223,4 @@ if filename_x is not None and filename_y is not None and layer is not None:
         btn = st.download_button(label="Download plot", data=file, file_name="plot.pdf", mime="image/pdf")
 
     print('8. End', datetime.now().strftime('%d-%m-%Y %H:%M:%S')) # test
-    print('9. RAM memory used:', round(psutil.virtual_memory().available / (1024.0 ** 3), 3), 'GB') # test
+    print(f"9. RAM used: {psutil.Process().memory_info().rss / (1024 * 1024):.2f} MB") # test