Update app.py

app.py

@@ -56,7 +56,7 @@ st.pyplot(fig)
 if st.button('Magic'):
     loss = []
     n_splits=5
-    opts = ['Linear regressor', 'Decision Tree regressor', 'SVR']
+    opts = ['LinearRegressor', 'DecisionTreeRegressor', 'SVR']
     for opt in opts:
         kf = KFold(n_splits=n_splits, shuffle=True, random_state=32)
         cv_scores = []
@@ -77,7 +77,7 @@ if st.button('Magic'):
     plt.scatter(X_test[:,0], y_test)
     st.pyplot(fig)
 
-options = ['Linear regressor', 'Decision Tree regressor', 'SVR']
+options = ['LinearRegressor', 'DecisionTreeRegressor', 'SVR']
 model_type = st.selectbox('Select model type to use:', options)
 options = ['boosting', 'bagging', 'gradient descent']
 ensemble_type = st.selectbox('Select the ensemble type:', options)