| import streamlit as st | |
| import joblib | |
| from preprocessing import data_preprocessing | |
| # Load your trained BoW model and vectorizer | |
| vectorizer_path = 'model_data/bow_vectorizer.joblib' | |
| model_path = 'model_data/bow_model.joblib' | |
| vectorizer = joblib.load(vectorizer_path) | |
| model = joblib.load(model_path) | |
| # Streamlit UI | |
| def predict(input): | |
| processed_text = data_preprocessing(input) | |
| user_input_bow = vectorizer.transform([processed_text]) | |
| # Make a prediction | |
| prediction = model.predict(user_input_bow) | |
| return prediction | |
| # User text input |