Update app.py

app.py

@@ -40,7 +40,7 @@ def train_predict(symbol, start_date, end_date, forecast_days):
     models = {
         'XGBoost': xgb.XGBRegressor(),
         'LightGBM': lgb.LGBMRegressor(),
-    #    'CatBoost': cb.CatBoostRegressor(verbose=0),
+        'CatBoost': catboost.CatBoostRegressor(verbose=0),
         'HistGradientBoosting': HistGradientBoostingRegressor()
     }
     
@@ -79,7 +79,7 @@ def plot_results(df, end_date, forecast_days, results):
     plt.close()
 
     # Plot future predictions
-    plt.figure(figsize=(28, 14))
+    plt.figure(figsize=(14, 7))
     future_dates = [end_date + timedelta(days=i) for i in range(1, forecast_days + 1)]
     
     for model, result in results.items():