Add: requirements.txt

baseline_wander_removal.py

@@ -1,6 +1,6 @@
 import numpy as np
 from scipy.signal import medfilt
-import matplotlib.pyplot as plt
+# import matplotlib.pyplot as plt
 # import seaborn as sns
 
 # sns.set_theme()
@@ -80,19 +80,3 @@ def band_pass_filter(signal):
     # result = result/max_val
 
     return result
-
-
-if __name__ == '__main__':
-    signal = np.loadtxt("./Lead 2/n1_lead2.txt")[:1000]
-    signal_bs = bw_remover(BASIC_SRATE, signal)
-    signal_flt = band_pass_filter(signal_bs)
-    plt.figure(figsize=(16, 9))
-    plt.subplot(2, 1, 1)
-    plt.plot(signal)
-    plt.grid(True)
-    plt.title("RAW signal")
-    plt.subplot(2, 1, 2)
-    plt.plot(signal_flt)
-    plt.title("baseline removed signal")
-    plt.grid(True)
-    plt.show()

requirements.txt

@@ -0,0 +1,5 @@
+torch
+gradio
+scipy
+numpy
+pywavelets