Update app.py

app.py

@@ -1,14 +1,21 @@
 import gradio as gr
 import pickle
+import sklearn  # Ensure sklearn is imported
 
-# Load the trained model and vectorizer
-model = pickle.load(open('spam_model.pkl', 'rb'))
-vectorizer = pickle.load(open('vectorizer.pkl', 'rb'))
+# Load the trained model and vectorizer safely
+try:
+    model = pickle.load(open('model.pkl', 'rb'))  # Changed filename
+    vectorizer = pickle.load(open('vectorizer.pkl', 'rb'))  # Keep this as it is
+except Exception as e:
+    print(f"Error loading model: {e}")
 
 def predict_sms(message):
-    transformed_text = vectorizer.transform([message])
-    prediction = model.predict(transformed_text)[0]
-    return "Spam" if prediction == 1 else "Not Spam"
+    try:
+        transformed_text = vectorizer.transform([message])
+        prediction = model.predict(transformed_text)[0]
+        return "Spam" if prediction == 1 else "Not Spam"
+    except Exception as e:
+        return f"Error: {e}"
 
 # Gradio Web Interface
 iface = gr.Interface(