jarpan03 commited on
Commit
75d42c3
·
verified ·
1 Parent(s): 6f2ff6e

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,5 +40,5 @@ classification_threshold = 0.45
40
  if st.button("Predict"):
41
  prob = model.predict_proba(input_data)[0,1]
42
  pred = int(prob >= classification_threshold)
43
- result = "will need the engine maintenance" if pred == 1 else "maintenance not needed"
44
  st.write(f"Prediction: Engine {result}")
 
40
  if st.button("Predict"):
41
  prob = model.predict_proba(input_data)[0,1]
42
  pred = int(prob >= classification_threshold)
43
+ result = "will need maintenance!" if pred == 1 else "doesn't need any maintenance!"
44
  st.write(f"Prediction: Engine {result}")