Update app.py

app.py

@@ -8,14 +8,6 @@ import tensorflow as tf
 import os
 import numpy as np
 
-path = "./"
-checkpoint_path = os.path.join(path,"model.ckpt")
-model = Sequential()
-model.add(Dense(64, input_shape=(4000,), activation='relu'))
-model.add(Dense(8, activation='relu'))
-model.add(Dense(1, activation='sigmoid'))
-model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy'])
-model.load_weights(checkpoint_path)
 
 
 dataset = load_dataset("lhoestq/demo1")
@@ -28,15 +20,23 @@ st.sidebar.markdown(""" # **Step 2: 開始分析**""")
 
 def callback():
     data = load_wave()
-    st.line_chart(data)
-    st.line_chart(data.T[1])
-    '''
+    #st.line_chart(data)
+
+    path = "./"
+    checkpoint_path = os.path.join(path,"model.ckpt")
+    model = Sequential()
+    model.add(Dense(64, input_shape=(4000,), activation='relu'))
+    model.add(Dense(8, activation='relu'))
+    model.add(Dense(1, activation='sigmoid'))
+    model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy'])
+    model.load_weights(checkpoint_path)
+
     out = tf.round(model.predict(np.array(data.T[1]))).cpu()
+    
     if np.array(out)[0] == "0":
         st.text("Relax")
     else:
         st.text("Activate")
-    '''
     
 
 bt1 = st.button(