Update app.py

app.py

@@ -42,7 +42,7 @@ def model_prediction(age, height, weight, duration, heart_rate, body_temp, gende
             model = load_xgb()
         inputs = pd.DataFrame([{'Gender': gender, 'Age': age, 'Height': height, 'Weight': weight, 'Duration': duration, 'Heart_Rate': heart_rate, 'Body_Temp': body_temp}])
         predictions = model.predict(inputs)
-        st.write("""# Prediction : """ , predictions[0])
+        st.write("""# Prediction : """ , predictions)
 
 with st.sidebar:
     st.markdown("<h2 style='text-align: center; color: red;'>Settings Tab</h2>", unsafe_allow_html=True)