Campfireman commited on
Commit
6674317
1 Parent(s): 4506881

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,7 +10,7 @@ project = hopsworks.login()
10
  fs = project.get_feature_store()
11
 
12
  mr = project.get_model_registry()
13
- model = mr.get_model("titanic_modal_more_specs", version=1)
14
  model_dir = model.download()
15
  model = joblib.load(model_dir + "/titanic_model.pkl")
16
 
@@ -26,7 +26,7 @@ def tb_titanic(pclass,sex,age,sibsp,parch,embarked,fare_per_customer,cabin):
26
  input_list.append(cabin)
27
  # 'res' is a list of predictions returned as the label.
28
  #global res
29
- res = model.predict(np.asarray(input_list).reshape(8, 1))
30
  return "This guy will"
31
 
32
  demo = gr.Interface(
 
10
  fs = project.get_feature_store()
11
 
12
  mr = project.get_model_registry()
13
+ model = mr.get_model("titanic_modal_more_specs", version=2)
14
  model_dir = model.download()
15
  model = joblib.load(model_dir + "/titanic_model.pkl")
16
 
 
26
  input_list.append(cabin)
27
  # 'res' is a list of predictions returned as the label.
28
  #global res
29
+ res = model.predict(np.asarray(input_list).reshape(1, 8))
30
  return "This guy will"
31
 
32
  demo = gr.Interface(