Upload folder using huggingface_hub
Browse files
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
|
| 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}")
|