Vishwas1 commited on
Commit
b2a4427
·
verified ·
1 Parent(s): 70da8f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -84,7 +84,8 @@ def forecast_ticker(ticker: str,
84
  feat_dynamic_real_dim=0,
85
  past_feat_dynamic_real_dim=0,
86
  )
87
- predictor = model.create_predictor(batch_size=32, device=DEVICE)
 
88
 
89
  # 4) Predict
90
  forecast = next(iter(predictor.predict(ds)))
 
84
  feat_dynamic_real_dim=0,
85
  past_feat_dynamic_real_dim=0,
86
  )
87
+ predictor = model.create_predictor(batch_size=32) # remove device=...
88
+
89
 
90
  # 4) Predict
91
  forecast = next(iter(predictor.predict(ds)))