Update app.py

app.py

@@ -6,7 +6,10 @@ from sklearn.preprocessing import PolynomialFeatures
 from sklearn.metrics import mean_squared_error
 
 st.title("Ridge Demo")
-degree = st.slider('Degree', 2, 40, 1)
+col1, col2 = st.columns(2)
+
+with col1:
+    degree = st.slider('Degree', 2, 40, 1)
 x = np.linspace(-1., 1., 100)
 y = 4 + 3*x + 2*np.sin(x) + 2*np.random.randn(len(x))
 
@@ -36,7 +39,8 @@ ax.set_ylabel("y")
 rmse = np.round(np.sqrt(mean_squared_error(y_pred, y)), 2)
 ax.set_title(f"Train RMSE: {rmse}")
 
-st.pyplot(fig)
+with col1:
+    st.pyplot(fig)
 hide_streamlit_style = """
             <style>
             #MainMenu {visibility: hidden;}