Update app.py

app.py

@@ -1,10 +1,13 @@
 import streamlit as st
 import pandas as pd
+import pickle
 from sklearn.naive_bayes import GaussianNB
 import joblib
 
 # Load the trained model
-model = joblib.load('naive_bayes_model.pkl')
+# Load the trained model using pickle
+with open('naive_bayes_model.pkl', 'rb') as file:
+    model = pickle.load(file)
 
 # Define the Streamlit app
 def main():