ricardo-lsantos commited on
Commit
7c2618e
1 Parent(s): 9cecb44

Added Download button for the model

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -15,6 +15,7 @@ def runNN(epoch, alpha):
15
 
16
  print(modelo)
17
  # Save model to file
 
18
  save_model(modelo, "modelo.json")
19
 
20
  for i in range(4):
 
15
 
16
  print(modelo)
17
  # Save model to file
18
+ st.download_button("Download model", data=modelo, file_name="modelo.json", mime="text/json")
19
  save_model(modelo, "modelo.json")
20
 
21
  for i in range(4):