Update app.py

app.py

@@ -7,7 +7,7 @@ import requests
 import json
 import dateutil.parser as dp
 import pandas as pd
-
+from huggingface_hub import hf_hub_url, cached_download
 
 def get_row():
     response_tomtom = requests.get(
@@ -50,7 +50,9 @@ def get_row():
     row.dropna(axis=0, inplace=True)
     return row
 
-model = joblib.load("./traffic_model.pkl")
+model = joblib.load(cached_download(
+    hf_hub_url(tilos/Traffic_Prediction, traffic_model.pkl)
+))
 
 def infer(input_dataframe):
   return pd.DataFrame(model.predict(input_dataframe))