Spaces:
Runtime error
Runtime error
SevenhuijsenM
commited on
Commit
·
9f417d1
1
Parent(s):
f361f28
Use different model
Browse files
app.py
CHANGED
@@ -16,12 +16,11 @@ project = hopsworks.login()
|
|
16 |
fs = project.get_feature_store()
|
17 |
mr = project.get_model_registry()
|
18 |
|
19 |
-
model = mr.get_model("
|
20 |
model_dir = model.download()
|
21 |
-
model = joblib.load(model_dir + "/
|
22 |
print("Model downloaded")
|
23 |
|
24 |
-
|
25 |
def predict(code, magnitudeOfDelay, hour, iconCategory, latitude, longitude, month):
|
26 |
|
27 |
# Change the magnitude delay to an integer based on the index
|
|
|
16 |
fs = project.get_feature_store()
|
17 |
mr = project.get_model_registry()
|
18 |
|
19 |
+
model = mr.get_model("sthlm_incidents_model", version=3)
|
20 |
model_dir = model.download()
|
21 |
+
model = joblib.load(model_dir + "/sthlm_model.pkl")
|
22 |
print("Model downloaded")
|
23 |
|
|
|
24 |
def predict(code, magnitudeOfDelay, hour, iconCategory, latitude, longitude, month):
|
25 |
|
26 |
# Change the magnitude delay to an integer based on the index
|