Ziyou Li commited on
Commit
8f4f9a0
·
1 Parent(s): c45f40d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -59,9 +59,7 @@ model = joblib.load(cached_download(
59
  ))
60
 
61
  def infer(input_dataframe):
62
- ts = datetime.timestamp(input_dataframe["referenceTime"])
63
- print(ts)
64
- input_dataframe["referenceTime"] = ts
65
  return pd.DataFrame(model.predict(input_dataframe)).clip(0, 1)
66
 
67
  title = "Stoclholm Highway E4 Real Time Traffic Prediction"
 
59
  ))
60
 
61
  def infer(input_dataframe):
62
+ input_dataframe["referenceTime"] = input_dataframe["referenceTime"].to_timestamp()
 
 
63
  return pd.DataFrame(model.predict(input_dataframe)).clip(0, 1)
64
 
65
  title = "Stoclholm Highway E4 Real Time Traffic Prediction"