Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 import numpy as np
+import time
 import matplotlib.pyplot as plt
 from sklearn.linear_model import LinearRegression
 from sklearn.metrics import mean_squared_error
@@ -41,7 +42,7 @@ if mode != st.session_state.current_mode:
 if train_btn:
 
     with st.spinner("⏳ Training model..."):
-        st.sleep(0.5)
+        time.sleep(0.5)
 
         if mode == "2D Regression":
             X = np.linspace(0, 10, num_points).reshape(-1, 1)