fix download button

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import json
 from nn import train, predict, save_model, sigmoid
 
 # INPUTS = [[0,0],[0,1],[1,0],[1,1]]
@@ -12,11 +13,15 @@ OUTPUTS = [[0],[1],[1],[0]]
 def runNN(epoch, alpha):
     # Train model
     modelo = train(epochs=epoch, alpha=alpha)
-    
     print(modelo)
     # Save model to file
-    st.download_button("Download model", data=modelo, file_name="modelo.json", mime="text/json")
-    save_model(modelo, "modelo.json")
+    # save_model(modelo, "modelo.json")
+    st.download_button(
+        label="Download model",
+        data=json.dumps(modelo),
+        file_name="modelo.json",
+        mime="application/json",
+    )
 
     for i in range(4):
         result = predict(INPUTS[i][0],INPUTS[i][1], activation=sigmoid)