Update app.py

app.py

@@ -3,6 +3,17 @@ import numpy as np
 import pickle
 import streamlit.components.v1 as components
 from sklearn.preprocessing import LabelEncoder
+import os
+
+if os.path.exists('label_encoder.pkl'):
+    try:
+        with open('label_encoder.pkl', 'rb') as f:
+            le = pickle.load(f)
+    except Exception as e:
+        print(f"Error loading the label encoder: {e}")
+else:
+    print("File 'label_encoder.pkl' does not exist.")
+
 
 # Load the pickled model
 def load_model():