Collapseruin commited on
Commit
0d108b2
·
1 Parent(s): 23e29f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,12 +10,12 @@ fs = project.get_feature_store()
10
 
11
 
12
  mr = project.get_model_registry()
13
- model_red = mr.get_model("wine_red_model", version=2)
14
  model_dir_red = model_red.download()
15
  model_red = joblib.load(model_dir + "/wine_red_model.pkl")
16
  print("Red Model downloaded")
17
 
18
- model_white = mr.get_model("wine_white_model", version=2)
19
  model_dir_white = model_white.download()
20
  model_white = joblib.load(model_dir_white + "/wine_white_model.pkl")
21
  print("White Model downloaded")
 
10
 
11
 
12
  mr = project.get_model_registry()
13
+ model_red = mr.get_model("wine_red_model", version=1)
14
  model_dir_red = model_red.download()
15
  model_red = joblib.load(model_dir + "/wine_red_model.pkl")
16
  print("Red Model downloaded")
17
 
18
+ model_white = mr.get_model("wine_white_model", version=1)
19
  model_dir_white = model_white.download()
20
  model_white = joblib.load(model_dir_white + "/wine_white_model.pkl")
21
  print("White Model downloaded")