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
|