diff --git a/training.py b/training.py
index 1cf28f9..8a0f3ab 100644
--- a/training.py
+++ b/training.py
@@ -44,8 +44,8 @@ else:
                     optimizer=tf.keras.optimizers.Adam(0.001),
                     metrics=[tf.keras.metrics.RootMeanSquaredError()])
 
-    verbose = 0 if len(verbose) == 0 else verbose[0]
-    epochs = 100 if len(epochs) == 0 else epochs[0]
+    verbose = 0 if len(verbose) == 0 else int(verbose[0])
+    epochs = 100 if len(epochs) == 0 else int(epochs[0])
     model.fit(train_X, train_Y, verbose=verbose, epochs=epochs)
 
     model.save('linear_regression.h5')
\ No newline at end of file