Update app.py

app.py

@@ -53,8 +53,7 @@ model.add(Dense(16, activation='relu'))
 model.add(Dense(32, activation='relu'))
 model.add(Dense(len(np.unique(y_train)), activation='softmax'))
 model.compile(loss='sparse_categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
-
- if st.button("Train Model"):
+if st.button("Train Model"):
     history = model.fit(X_train, y_train, validation_split=0.2, epochs=epochs, batch_size=batch_size, verbose=0)
     st.success("Model training complete.")