Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -122,7 +122,7 @@ def main():
|
|
122 |
st.write(user_input)
|
123 |
|
124 |
prediction = model.predict(user_input)
|
125 |
-
prediction_proba = model.
|
126 |
|
127 |
st.subheader('Prediction:')
|
128 |
if prediction[0] == 1:
|
|
|
122 |
st.write(user_input)
|
123 |
|
124 |
prediction = model.predict(user_input)
|
125 |
+
prediction_proba = model.predict(user_input, raw_score=True)
|
126 |
|
127 |
st.subheader('Prediction:')
|
128 |
if prediction[0] == 1:
|