SevenhuijsenM commited on
Commit
9f417d1
·
1 Parent(s): f361f28

Use different model

Browse files
Files changed (1) hide show
  1. app.py +2 -3
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("stockholm_incidents_model", version=1)
20
  model_dir = model.download()
21
- model = joblib.load(model_dir + "/stckhlm_inc_model.pkl")
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