Update app.py

app.py

@@ -62,7 +62,14 @@ def infer(input_dataframe):
     serie = input_dataframe["referenceTime"]
     ts = dp.parse(serie.iloc[0]).timestamp()
     input_dataframe["referenceTime"] = ts
-    return pd.DataFrame(model.predict(input_dataframe)).clip(0, 1)
+    res = pd.DataFrame(model.predict(input_dataframe)).clip(0, 1)
+    if res > 0.8:
+        status = "Smooth Traffic on E4"
+    elif res > 0.5:
+        status = "Slight congestion on E4"
+    else:
+        status = "Total congestion on E4"
+    return res, status
 
 title = "Stoclholm Highway E4 Real Time Traffic Prediction"
 description = "Stockholm E4 (59°23'44.7"" N 17°59'00.4""E) highway real time traffic prediction"