Hem345 commited on
Commit
5ffd414
·
verified ·
1 Parent(s): 3bd9c33

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -43,6 +43,7 @@ model_path = "mnist_cnn_model.h5"
43
  if st.button("Train Model"):
44
  model = create_model()
45
  with st.spinner("Training..."):
 
46
  history = model.fit(train_images, train_labels, validation_data=(test_images, test_labels), epochs=10, batch_size=64)
47
 
48
  # Save the model
 
43
  if st.button("Train Model"):
44
  model = create_model()
45
  with st.spinner("Training..."):
46
+ st.text("Training now:")
47
  history = model.fit(train_images, train_labels, validation_data=(test_images, test_labels), epochs=10, batch_size=64)
48
 
49
  # Save the model