Views
No views yet
1import pickle
2import requests
3
4# Load the model and vectorizer
5model = pickle.load(open('model.pkl', 'rb'))
6vectorizer = pickle.load(open('vectorizer.pkl', 'rb'))
7preprocess_func = pickle.load(open('preprocess.pkl', 'rb'))
8
9# Preprocess and predict
10def predict(text):
11 processed_text = preprocess_func(text)
12 vectorized_text = vectorizer.transform([processed_text])
13 prediction = model.predict(vectorized_text)[0]
14 return prediction