Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|