ayush-vatsal commited on
Commit
2503d1b
1 Parent(s): 1bbd7db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -12,10 +12,9 @@ st.set_page_config(
12
  page_icon='',
13
  layout='wide'
14
  )
15
- project = data
16
 
17
  @st.cache(allow_output_mutation=True,suppress_st_warning=True)
18
- def get_model(project = project):
19
  mr = project.get_model_registry()
20
  model = mr.get_model("cc_fraud", version = 1)
21
  model_dir = model.download()
@@ -94,7 +93,7 @@ if (input_list[13] == "Manual"):
94
  input_list.insert(14, 1)
95
 
96
  df = pd.DataFrame(input_list)
97
- model = get_model(project)
98
  res = model.predict(df.T)[0].round(4)
99
 
100
  with model_train:
 
12
  page_icon='',
13
  layout='wide'
14
  )
 
15
 
16
  @st.cache(allow_output_mutation=True,suppress_st_warning=True)
17
+ def get_model():
18
  mr = project.get_model_registry()
19
  model = mr.get_model("cc_fraud", version = 1)
20
  model_dir = model.download()
 
93
  input_list.insert(14, 1)
94
 
95
  df = pd.DataFrame(input_list)
96
+ model = get_model()
97
  res = model.predict(df.T)[0].round(4)
98
 
99
  with model_train: