Spaces:

saad177
/

Diabetes-Prediction

Runtime error

saad177 commited on Jan 2

Commit

89d1951

•

1 Parent(s): 350dd14

update to the latest model, gan v2

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ print("trying to dl model")
 mr = project.get_model_registry()
 model = mr.get_model("diabetes_gan_model", version=2)
 model_dir = model.download()
-model = joblib.load(model_dir + "/diabetes_model.pkl")
 print("Model downloaded")
 diabetes_fg = fs.get_feature_group(name="diabetes_gan", version=1)

 mr = project.get_model_registry()
 model = mr.get_model("diabetes_gan_model", version=2)
 model_dir = model.download()
+model = joblib.load(model_dir + "/diabetes_gan_model.pkl")
 print("Model downloaded")
 diabetes_fg = fs.get_feature_group(name="diabetes_gan", version=1)