Update app.py

app.py

@@ -33,6 +33,21 @@ def classify_image(image):
         st.error("Image classification failed. Please try again.")
         return None
 
+@st.cache_data
+def load_datasets():
+    try:
+        with st.spinner('Loading dataset...'):
+            # Use BytesIO to read the CSV content
+            original_data = pd.read_csv('CTP_Model1.csv', low_memory=False)
+            
+            
+            # Ensure column names match the model's expectations
+            original_data.columns = original_data.columns.str.strip().str.capitalize()
+            return original_data
+    except Exception as e:
+        st.error(f"Error loading dataset: {str(e)}")
+        raise e
+
 def find_closest_match(df, brand, model):
     match = df[(df['make'].str.contains(brand, case=False)) & (df['model'].str.contains(model, case=False))]
     if not match.empty: