model path changed update

src/model.py

@@ -8,7 +8,7 @@ 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
-
+import pickle
 # at beginning of the script
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
@@ -38,11 +38,9 @@ class Model_Load:
     def store_model_load(self,model_option):
         if model_option=='TFT':
           # best_model_path="models/store_item_10_lead_1_v2/lightning_logs/lightning_logs/version_2/checkpoints/epoch=7-step=4472.ckpt"
-          best_model_path="models/store_item_10_lead_1_v3/lightning_logs/lightning_logs/version_0/checkpoints/epoch=7-step=4472.ckpt"
-          best_tft = TemporalFusionTransformer.load_from_checkpoint(best_model_path)
-          # best_tft = TemporalFusionTransformer()
-          # best_tft.load_state_dict(torch.load(best_model_path,map_location=torch.device('cpu')))
-          # best_tft.to('cpu')
+          # best_model_path="models/store_item_10_lead_1_v3/lightning_logs/lightning_logs/version_0/checkpoints/epoch=7-step=4472.ckpt"
+          # best_tft = TemporalFusionTransformer.load_from_checkpoint(best_model_path)
+          best_tft=pickle.load(open("models/cpu_finalized_model_v1.sav", 'rb'))
           print('Model Load Sucessfully.')
           return best_tft
         elif model_option=='Prophet':