Update prediction.py

prediction.py

@@ -57,7 +57,7 @@ def run():
     if submitted:        
         y_pred_inf = model.predict(df)
         if y_pred_inf[0] == 0:
-            st.subheader('~ This Customer is Predicted Not to Churn ~')
+            st.write('~ This Customer is Predicted Not to Churn ~')
         else:
             st.write('~ This Customer is Predicted to Churn ~')