Update app.py

app.py

@@ -18,16 +18,17 @@ import nltk
 nltk.download('stopwords')
 nltk.download('punkt')
 
+# Load the trained model
 model = joblib.load('model.bin')
 
-# Initialize the vectorizer
-vectorizer = CountVectorizer()
+# Load the pre-trained vectorizer
+vectorizer = joblib.load('vectorizer.bin')
 
 def remove_punctuation(text):
     punctuation_free = "".join([i for i in text if i not in string.punctuation])
     return punctuation_free
 
-def test_model(text, vectorizer):
+def test_model(text):
     # Convert text to lowercase
     text = text.lower()