Update app.py

app.py

@@ -1,5 +1,3 @@
-# app.py
-
 import streamlit as st
 import pandas as pd
 import numpy as np
@@ -60,12 +58,12 @@ elif selection == "Make a Prediction":
 
 
     submit_button = st.form_submit_button("Predict")
-
+    
     if submit_button:
         # Construct the input DataFrame
-        input_features = pd.DataFrame([[sub_category, sales, quantity, discount
+        input_features = pd.DataFrame([[ sales, discount, quantity, sub_category
         ]], columns=[
-            'Sub-Category', 'Product Name', 'Sales', 'Quantity', 'Discount'
+            'Sales', 'Discount', 'Quantity', 'Sub-Category'
         ])
         
         # Preprocess and predict (You'll need to adjust this part based on how your model expects input)