Update app.py

app.py

@@ -62,14 +62,14 @@ def infer(input_dataframe):
     serie = input_dataframe["referenceTime"]
     ts = dp.parse(serie.iloc[0]).timestamp()
     input_dataframe["referenceTime"] = ts
-    res = pd.DataFrame(model.predict(input_dataframe)).clip(0, 1)
+    res = pd.DataFrame(model.predict(input_dataframe)).clip(0, 1).iloc[0, 0]
     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 pd.Dataframe[res, status]
+    return pd.Dataframe({'Freeflow Level':[res], 'Status': [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"