Update streamlit_simulation/app.py

streamlit_simulation/app.py

@@ -134,7 +134,11 @@ def load_transformer_model_and_dataset():
 
     # Load model
     model = load_moment_model()
-    checkpoint_path = os.path.join(CHECKPOINT_DIR, "model_final.pth")
+    checkpoint_path = hf_hub_download(
+    repo_id="dlaj/energy-forecasting-files",
+    filename="transformer_model/model_final.pth",
+    repo_type="dataset"
+    )
     model.load_state_dict(torch.load(checkpoint_path, map_location=device))
     model.to(device)
     model.eval()
@@ -148,7 +152,12 @@ def load_transformer_model_and_dataset():
 
 @st.cache_data
 def load_data():
-    df = pd.read_csv(DATA_PATH, parse_dates=["date"])
+    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"])
     return df