Update app.py

app.py

@@ -4,7 +4,7 @@ import numpy as np
 import os
 
 # Define Model File Path
-MODEL_PATH = "random_forest_pipeline.pkl"
+model_path = "random_forest_pipeline.pkl"
 
 # Streamlit UI Setup
 st.set_page_config(page_title="Wine Quality Prediction 🍷🔬", layout="centered")
@@ -28,14 +28,15 @@ st.markdown(
 st.markdown('<h1 class="title">Wine Quality Prediction 🍷🔬</h1>', unsafe_allow_html=True)
 st.write("Predicting Wine Quality based on wine parameters.")
 
-# Load Model if Available
-if os.path.exists(MODEL_PATH):
-    with open(MODEL_PATH, "rb") as f:
+# Check if model exists
+if os.path.exists(model_path):
+    with open(model_path, "rb") as f:
         model = pickle.load(f)
+    st.success("✅ Model loaded successfully!")
     model_loaded = True
 else:
+    st.error(f"❌ Model file '{model_path}' not found! Please upload or train the model first.")
     model_loaded = False
-    st.error(f"❌ Model file '{MODEL_PATH}' not found! Please upload the trained model.")
 
 # User Inputs
 fixed_acidity = st.number_input("Fixed Acidity", min_value=4.6, max_value=15.9, value=7.6)