Ziyou Li commited on
Commit
6f71655
·
1 Parent(s): 98d9c54

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -55,7 +55,7 @@ model = joblib.load(cached_download(
55
  ))
56
 
57
  def infer(input_dataframe):
58
- return pd.DataFrame(model.predict(input_dataframe))
59
 
60
  title = "Stoclholm Highway E4 Real Time Traffic Prediction"
61
  description = "Stockholm E4 (59°23'44.7"" N 17°59'00.4""E) highway real time traffic prediction"
 
55
  ))
56
 
57
  def infer(input_dataframe):
58
+ return pd.DataFrame(model.predict(input_dataframe)).clip(0, 1)
59
 
60
  title = "Stoclholm Highway E4 Real Time Traffic Prediction"
61
  description = "Stockholm E4 (59°23'44.7"" N 17°59'00.4""E) highway real time traffic prediction"