Update functions.py

functions.py

@@ -35,7 +35,7 @@ def preprocess_text(text):
     lemmatized_clean = [word.translate(str.maketrans('', '', string.punctuation)) for word in lemmatized_words]
     return ' '.join(lemmatized_clean)
 
-def prediction(model, X):
+def prediction(X):
     model = load_model('best_model.keras')
     y_pred = model.predict(X)
     predictions = np.argmax(y_pred, axis=1)