Update app.py

app.py

@@ -108,14 +108,6 @@ def get_user_input():
 
 user_input, submit_button = get_user_input()
 
-quality_mapping = {
-    3: "Worst quality (3)",
-    4: "Low quality (4)",
-    5: "Medium-to-low quality (5)",
-    6: "Medium-to-high quality (6)",
-    7: "Good quality (7)",
-    8: "Best quality (8)"
-}
 
 # When the 'Submit' button is pressed, perform the prediction
 if submit_button:
@@ -123,14 +115,13 @@ if submit_button:
     prediction = pipe.predict(user_input)
     prediction_value = prediction[0]
 
-    prediction_text = quality_mapping[prediction_value]
-
     # Display the prediction
     st.header("Predicted Quality")
-    st.write(prediction_text)
+    st.write(prediction_value)
+    
 
-    st.markdown(
-    """
-    [See how this model was created on Kaggle: 🍷 Wine Quality - EDA, Prediction and Deploy](https://www.kaggle.com/code/lusfernandotorres/wine-quality-eda-prediction-and-deploy/notebook)
-    """
+st.markdown(
+"""
+[See how this model was created on Kaggle: 🍷 Wine Quality - EDA, Prediction and Deploy](https://www.kaggle.com/code/lusfernandotorres/wine-quality-eda-prediction-and-deploy/notebook)
+"""
 )