Yasaman commited on
Commit
571cd8f
·
1 Parent(s): f205c21

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ def air_quality(city,model_name):
24
  X = X.drop(X.columns[0],axis=1)
25
 
26
  mr = project.get_model_registry()
27
- model = mr.get_model(model_name, version=1)
28
  model_dir = model.download()
29
  model = joblib.load(model_dir + "/model.pkl")
30
 
 
24
  X = X.drop(X.columns[0],axis=1)
25
 
26
  mr = project.get_model_registry()
27
+ model = mr.get_model(model_name, version=2)
28
  model_dir = model.download()
29
  model = joblib.load(model_dir + "/model.pkl")
30