Update src/streamlit_app.py

src/streamlit_app.py

@@ -55,7 +55,7 @@ if "hf_clicked" not in st.session_state:
 # Config & paths
 # -------------------------
 st.set_page_config(page_title="Steel Authority of India Limited (MODEX)", layout="wide")
-plt.style.use("seaborn-v0_8-muted")
+plt.style.use("seaborn-muted")
 sns.set_palette("muted")
 sns.set_style("whitegrid")
 
@@ -458,7 +458,7 @@ with tabs[4]:
         return df_clean
 
     df = clean_entire_df(df)
-    st.caption("✅ Dataset cleaned globally — all numeric-like values converted safely.")
+    st.caption(" Dataset cleaned globally — all numeric-like values converted safely.")
 
     # --- Use Case Selection ---
     use_case = st.selectbox(
@@ -564,7 +564,8 @@ with tabs[4]:
         study = optuna.create_study(direction="maximize")
         study.optimize(obj, n_trials=n_trials, show_progress_bar=False)
         params = study.best_trial.params if study.trials else {}
-        model = RandomForestRegressor(random_state=42)
+        if fam == "RandomForest":
+            model = RandomForestRegressor(**study.best_trial.params, random_state=42)
         return {"family": fam, "model_obj": model, "best_params": params, "cv_score": study.best_value}
 
     # --- Run button ---