saad177 commited on
Commit
350dd14
1 Parent(s): 76f11b6

update to the latest model, gan v2

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -15,7 +15,7 @@ fs = project.get_feature_store()
15
 
16
  print("trying to dl model")
17
  mr = project.get_model_registry()
18
- model = mr.get_model("diabetes_model", version=1)
19
  model_dir = model.download()
20
  model = joblib.load(model_dir + "/diabetes_model.pkl")
21
  print("Model downloaded")
@@ -188,6 +188,7 @@ with gr.Blocks() as demo:
188
 
189
  ## save user's data in hopsworks
190
  if consent_input == "accept":
 
191
  user_data_fg = fs.get_or_create_feature_group(
192
  name="user_diabetes_data",
193
  version=1,
 
15
 
16
  print("trying to dl model")
17
  mr = project.get_model_registry()
18
+ model = mr.get_model("diabetes_gan_model", version=2)
19
  model_dir = model.download()
20
  model = joblib.load(model_dir + "/diabetes_model.pkl")
21
  print("Model downloaded")
 
188
 
189
  ## save user's data in hopsworks
190
  if consent_input == "accept":
191
+ print("user consented to save their data, now trying to save to hopsworks")
192
  user_data_fg = fs.get_or_create_feature_group(
193
  name="user_diabetes_data",
194
  version=1,