tryout no plot

streamlit_simulation/app.py

@@ -133,10 +133,11 @@ init_session_state()
 
 # ============================== Loaders ==============================
 
-@st.cache_data
-def load_lightgbm_model():
-    with open(MODEL_PATH_LIGHTGBM, "rb") as f:
-        return pickle.load(f)
+CSV_PATH_HF = hf_hub_download(
+repo_id="dlaj/energy-forecasting-files",
+filename="data/processed/energy_consumption_aggregated_cleaned.csv",
+repo_type="dataset"
+)
 
 @st.cache_resource
 def load_transformer_model_and_dataset():
@@ -152,15 +153,9 @@ def load_transformer_model_and_dataset():
     model.to(device)
     model.eval()
 
-    csv_path = hf_hub_download(
-    repo_id="dlaj/energy-forecasting-files",
-    filename="data/processed/energy_consumption_aggregated_cleaned.csv",
-    repo_type="dataset"
-    )
-
     # Datasets
-    train_dataset = InformerDataset(data_split="train", forecast_horizon=FORECAST_HORIZON, random_seed=13, csv_path=csv_path)
-    test_dataset = InformerDataset(data_split="test", forecast_horizon=FORECAST_HORIZON, random_seed=13, csv_path=csv_path)
+    train_dataset = InformerDataset(data_split="train", forecast_horizon=FORECAST_HORIZON, random_seed=13, csv_path=CSV_PATH_HF)
+    test_dataset = InformerDataset(data_split="test", forecast_horizon=FORECAST_HORIZON, random_seed=13, csv_path=CSV_PATH_HF)
     test_dataset.scaler = train_dataset.scaler
 
     return model, test_dataset, device
@@ -168,14 +163,15 @@ def load_transformer_model_and_dataset():
 
 @st.cache_data
 def load_data():
-    csv_path = hf_hub_download(
-        repo_id="dlaj/energy-forecasting-files",
-        filename="data/processed/energy_consumption_aggregated_cleaned.csv",
-        repo_type="dataset"
-    )
-    df = pd.read_csv(csv_path, parse_dates=["date"])
+    df = pd.read_csv(CSV_PATH_HF, parse_dates=["date"])
     return df
 
+#Load lightgbm model
+@st.cache_data
+def load_lightgbm_model():
+    with open(MODEL_PATH_LIGHTGBM, "rb") as f:
+        return pickle.load(f)
+
 
 # ============================== Utility Functions ==============================