Update app.py

app.py

@@ -6,36 +6,16 @@ import hopsworks
 import joblib
 import os
 
-project = hopsworks.login(api_key_value="0rdWXlLgEd3mkGOg.iRZ7TtAkWGPlJHNQcAEph6Qbokoaq7QTBRI9ckwWUki8tIYGyBvrKhJvtLoUOGQ4")
+project = hopsworks.login(api_key_value="PLzrFWpU9maWMoDr.T85eoNUrXUn2wHvZX0baV4Gp1wD1MbpaXI9ilBKtd0d2A5YPqui3nTrJPR3w9aYk")
 fs = project.get_feature_store()
 
-# mr = project.get_model_registry()
-# model = mr.get_model("xgboost_model", version=1)
-# model_dir = model.download()
-# model = joblib.load("/model.pkl")
+mr = project.get_model_registry()
+model = mr.get_model("xgboost_model", version=1)
+model_dir = model.download()
+model = joblib.load(model_dir+ '/weather_model.pkl')
+
 
 
-def get_model(project, model_name):
-    """Retrieve desired model or download it from the Hopsworks Model Registry.
-    In second case, it will be physically downloaded to this directory"""
-    
-    TARGET_FILE = "model.pkl"
-    list_of_files = [os.path.join(dirpath,filename) for dirpath, _, filenames \
-                     in os.walk('.') for filename in filenames if filename == TARGET_FILE]
-
-    if list_of_files:
-        model_path = list_of_files[0]
-        model = joblib.load(model_path)
-    else:
-        if not os.path.exists(TARGET_FILE):
-            mr = project.get_model_registry()
-            model = mr.get_model("xgboost_model", version=1)
-            model_dir = model.download()
-            model = joblib.load("/model.pkl")
-
-    return model
-
-model = get_model(project,"xgboost_model")
 
 def forecast():