bugfix

.ipynb_checkpoints/app-checkpoint.py

@@ -59,7 +59,7 @@ def predict(model, dataloader):
 ## Initiate Data
 with open('data/parameters.pkl', 'rb') as f:
     parameters = pickle.load(f)
-model = TemporalFusionTransformer.load_from_checkpoint('model/tft_check.ckpt')
+model = TemporalFusionTransformer.cpu().load_from_checkpoint('model/tft_check.ckpt')
 
 df = pd.read_pickle('data/test_data.pkl')
 df = df.loc[(df["Branch"] == 15) & (df["Group"].isin(["6","7","4","1"]))]

app.py

@@ -59,7 +59,7 @@ def predict(model, dataloader):
 ## Initiate Data
 with open('data/parameters.pkl', 'rb') as f:
     parameters = pickle.load(f)
-model = TemporalFusionTransformer.load_from_checkpoint('model/tft_check.ckpt')
+model = TemporalFusionTransformer.cpu().load_from_checkpoint('model/tft_check.ckpt')
 
 df = pd.read_pickle('data/test_data.pkl')
 df = df.loc[(df["Branch"] == 15) & (df["Group"].isin(["6","7","4","1"]))]