bugfix

.ipynb_checkpoints/app-checkpoint.py

@@ -54,7 +54,7 @@ def prepare_dataset(parameters, df, rain, temperature, datepicker):
     return df.to_dataloader(train=False, batch_size=256,num_workers = 0)
 
 def predict(model, dataloader): 
-    return model.predict(dataloader, mode="raw", return_x=True, return_index=True)
+    return model.cpu().predict(dataloader, mode="raw", return_x=True, return_index=True)
     
 ## Initiate Data
 with open('data/parameters.pkl', 'rb') as f:

app.py

@@ -54,7 +54,7 @@ def prepare_dataset(parameters, df, rain, temperature, datepicker):
     return df.to_dataloader(train=False, batch_size=256,num_workers = 0)
 
 def predict(model, dataloader): 
-    return model.predict(dataloader, mode="raw", return_x=True, return_index=True)
+    return model.cpu().predict(dataloader, mode="raw", return_x=True, return_index=True)
     
 ## Initiate Data
 with open('data/parameters.pkl', 'rb') as f: