Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -232,6 +232,7 @@ def predict(
|
|
232 |
# Output
|
233 |
predicted_status = "Unknown" if pred[0] == -1 else encoders['status_name'].inverse_transform([pred])[0]
|
234 |
probability = pred_proba[0][pred[0]] * 100 if pred[0] != -1 else "Unknown"
|
|
|
235 |
|
236 |
if predicted_status == "Returned to Client":
|
237 |
probability = 100 - probability
|
|
|
232 |
# Output
|
233 |
predicted_status = "Unknown" if pred[0] == -1 else encoders['status_name'].inverse_transform([pred])[0]
|
234 |
probability = pred_proba[0][pred[0]] * 100 if pred[0] != -1 else "Unknown"
|
235 |
+
print(predicted_status)
|
236 |
|
237 |
if predicted_status == "Returned to Client":
|
238 |
probability = 100 - probability
|