Update app.py

app.py

@@ -9,6 +9,7 @@ import dateutil.parser as dp
 import pandas as pd
 from huggingface_hub import hf_hub_url, cached_download
 import time
+import datetime as datetime
 
 def get_row():
     response_tomtom = requests.get(
@@ -38,7 +39,7 @@ def get_row():
     vis           = json_response_smhi["timeSeries"][0]["parameters"][9]["values"][0] # Visibility
 
     # Use current time
-    referenceTime = str(Timestamp(time.time()))
+    referenceTime = datetime.fromtimestamp(time.time())
     
     row           ={"referenceTime": referenceTime, 
                         "temperature": t, 
@@ -58,7 +59,7 @@ model = joblib.load(cached_download(
 ))
 
 def infer(input_dataframe):
-    input_dataframe["referenceTime"] = dp.parse(input_dataframe["referenceTime"]).timestamp()
+    input_dataframe["referenceTime"] = datetime.timestamp(input_dataframe["referenceTime"])
     return pd.DataFrame(model.predict(input_dataframe)).clip(0, 1)
 
 title = "Stoclholm Highway E4 Real Time Traffic Prediction"