Update src/model.py

src/model.py

@@ -7,7 +7,7 @@ import pandas as pd
 import torch
 from prophet.serialize import model_to_json, model_from_json
 from pytorch_forecasting import Baseline, TemporalFusionTransformer, TimeSeriesDataSet
-from pytorch_forecasting.models.temporal_fusion_transformer.tuning import optimize_hyperparameters
+# from pytorch_forecasting.models.temporal_fusion_transformer.tuning import optimize_hyperparameters
 import pickle
 # at beginning of the script
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")