Merge pull request #57 from statgarten/ktr

fix : time series forecasting frontend & backend
statgarten · Nov 15, 2023 · 66275df · 66275df
2 parents 3a8a60f + 02088ba
commit 66275df
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/fastapi_backend/server.py b/fastapi_backend/server.py
@@ -393,6 +393,7 @@ def additional_pred(model, input_sequense, num_features, scaler, device):
 # training 
 @app.post("/time_train")
 async def time_train_endpoint(data_arranges:list[str],
+                              file_ext: str = Form(...),
                               pred_col: str = Form(...),
                               date: str = Form(...),
                               window_size: int = Form(...),

diff --git a/streamlit_frontend/time_series_forecasting.py b/streamlit_frontend/time_series_forecasting.py
@@ -274,7 +274,9 @@ def main():
                     with open('timeseries_sample_data/timeseries_forecasting_sample_data.csv', mode = 'rb') as f:
                         csv_file_obj = io.BytesIO(f.read())
                         csv_files = {'file': csv_file_obj}
-
+
+                    file_ext = 'csv'
+
                 date_column, pred_column = st.columns(2)
 
                 with date_column:
@@ -449,4 +451,4 @@ def main():
 
 # For running this file individually
 # if __name__ == "__main__":
-#     app()
+#     app()