Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 =
|
29 |
-
prob =
|
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 |
|