Update app.py

app.py

@@ -5,19 +5,16 @@ import pickle
 
 print("\n--- Creating Gradio Interface ---")
 
-# Load the saved assets
-try:
-    with open('final_model.pkl', 'rb') as file:
-        final_model = pickle.load(file)
-    with open('scaler.pkl', 'rb') as file:
-        scaler = pickle.load(file)
-    with open('label_encoder.pkl', 'rb') as file:
-        label_encoder = pickle.load(file)
-    print("✓ Models, Scaler, and Label Encoder loaded successfully.")
-except Exception as e:
-    print(f"✗ Error loading saved assets: {e}")
-    print("Please ensure 'final_model.pkl', 'scaler.pkl', and 'label_encoder.pkl' are in the current directory.")
-    exit()
+file = 'final_model.pkl'
+final_model = pickle.load(file)
+
+file = 'scaler.pkl'
+scaler = pickle.load(file)
+
+file = 'label_encoder.pkl'
+label_encoder = pickle.load(file)
+
+print("✓ Models, Scaler, and Label Encoder loaded successfully.")
 
 # Get original feature names from X (from previous execution, assuming it's available or re-derivable)
 # If X is not in kernel state, we'd need to load the original dataset and derive column names