Update app.py

app.py

@@ -98,5 +98,9 @@ if st.button("Submit"):
     prediction = model.predict(df_encoded_array)
     
     st.write(f"SpaceX Stage 1 Predictor will be : {classes[prediction[0]]}")
+    if prediction[0] == 1 :
+        st.image('success.gif')
+    elif prediction[0] == 0 :
+        st.image('failed.gif')