Update app.py

app.py

@@ -11,7 +11,7 @@ fs = project.get_feature_store()
 
 mr = project.get_model_registry()
 model = mr.get_model("xgboost_model", version=1)
-# model_dir = model.download()
+model_dir = model.download()
 # model = joblib.load(model_dir + "/model.pkl")
 
 
@@ -26,7 +26,7 @@ def forecast():
 
     res = model.predict(np.asarray(x).reshape(-1, 1))
     
-    return res
+    return model_dir
 
 demo = gr.Interface(
     fn=forecast,