Update app.py
Browse files
app.py
CHANGED
@@ -97,6 +97,6 @@ if st.button("Submit"):
|
|
97 |
classes = ['Failed','Succeed']
|
98 |
prediction = model.predict(df_encoded_array)
|
99 |
|
100 |
-
st.write(f"SpaceX Stage 1 Predictor will be : {classes[prediction]}")
|
101 |
|
102 |
|
|
|
97 |
classes = ['Failed','Succeed']
|
98 |
prediction = model.predict(df_encoded_array)
|
99 |
|
100 |
+
st.write(f"SpaceX Stage 1 Predictor will be : {classes[prediction[0]]}")
|
101 |
|
102 |
|