Spaces:
Runtime error
Runtime error
Commit
·
0d108b2
1
Parent(s):
23e29f0
Update app.py
Browse files
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=
|
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=
|
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")
|