bugfix

app.py

@@ -31,7 +31,7 @@ def raw_preds_to_df(raw,quantiles = None):
     preds_df = pd.DataFrame(index.values.repeat(dec_len * n_quantiles, axis=0),columns=index.columns)
     preds_df = preds_df.assign(h=np.tile(np.repeat(np.arange(1,1+dec_len),n_quantiles),len(preds_df)//(dec_len*n_quantiles)))
     preds_df = preds_df.assign(q=np.tile(np.arange(n_quantiles),len(preds_df)//n_quantiles))
-    preds_df = preds_df.assign(pred=preds.flatten().cpu().numpy())
+    preds_df = preds_df.assign(pred=preds.flatten().numpy())
     if quantiles is not None:
         preds_df['q'] = preds_df['q'].map({i:q for i,q in enumerate(quantiles)})