Update app.py

app.py

@@ -2,7 +2,6 @@ import gradio as gr
 import numpy as np
 from PIL import Image
 import requests
-from functions import decode_features, get_model
 import hopsworks
 import joblib
 
@@ -13,10 +12,29 @@ fs = project.get_feature_store()
 # model = mr.get_model("xgboost_model", version=1)
 # model_dir = model.download()
 # model = joblib.load("/model.pkl")
-model = get_model(project=project,
-                  model_name="xgboost_model",
-                  evaluation_metric="f1_score",
-                  sort_metrics_by="max")
+
+
+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():