Robin Goran commited on
Commit
22d8bb2
·
1 Parent(s): ae1a0dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,9 +10,9 @@ fs = project.get_feature_store()
10
 
11
 
12
  mr = project.get_model_registry()
13
- model = mr.get_model("wine_model", version=1)
14
  model_dir = model.download()
15
- model = joblib.load(model_dir + "/wine_model.pkl")
16
  print("Model downloaded")
17
 
18
  def wine_predictor(fixed_acidity, volatile_acidity, citric_acid, residual_sugar,
 
10
 
11
 
12
  mr = project.get_model_registry()
13
+ model = mr.get_model("wine_model_randomforest", version=1)
14
  model_dir = model.download()
15
+ model = joblib.load(model_dir + "/wine_model_randomforest.pkl")
16
  print("Model downloaded")
17
 
18
  def wine_predictor(fixed_acidity, volatile_acidity, citric_acid, residual_sugar,