pierrelf commited on
Commit
1ddb0f2
·
1 Parent(s): ada7da5

Update wine_model retrieval in app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ project = hopsworks.login()
7
  fs = project.get_feature_store()
8
 
9
  mr = project.get_model_registry()
10
- model = mr.get_model("wine_model", version=1)
11
  model_dir = model.download()
12
  model = joblib.load(model_dir + "/wine_model.pkl")
13
  print("Model downloaded")
 
7
  fs = project.get_feature_store()
8
 
9
  mr = project.get_model_registry()
10
+ model = mr.get_model("wine_model")
11
  model_dir = model.download()
12
  model = joblib.load(model_dir + "/wine_model.pkl")
13
  print("Model downloaded")