Nishthap commited on
Commit
4b00087
1 Parent(s): d7688b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,8 +25,8 @@ def predict_placement(data):
25
  new_data = pd.DataFrame(data, index=[0])
26
 
27
  # Make prediction
28
- prediction = model.predict(new_data)[0]
29
- prob = model.predict_proba(new_data)[0][1]
30
 
31
  return prediction, prob
32
 
 
25
  new_data = pd.DataFrame(data, index=[0])
26
 
27
  # Make prediction
28
+ prediction = model_campus.predict(new_data)[0]
29
+ prob = model_campus.predict_proba(new_data)[0][1]
30
 
31
  return prediction, prob
32