Views
No views yet
1from huggingface_hub import hf_hub_download
2import joblib
3
4# Download model files
5model_path = hf_hub_download(repo_id="21f1000330/sentiment-analysis-movie-reviews", filename="sentiment_model.pkl")
6vectorizer_path = hf_hub_download(repo_id="21f1000330/sentiment-analysis-movie-reviews", filename="sentiment_vectorizer.pkl")
7
8# Load model and vectorizer
9model = joblib.load(model_path)
10vectorizer = joblib.load(vectorizer_path)
11
12# Preprocess text (same as training)
13def preprocess_text(text):
14 from bs4 import BeautifulSoup
15 import re
16 text = BeautifulSoup(text, "html.parser").get_text() if text else ""
17 text = re.sub(r'[^a-zA-Z0-9\s]', '', text) if text else ""
18 return text.lower().strip()
19
20# Make prediction
21text = "This movie was amazing! I loved it."
22preprocessed = preprocess_text(text)
23text_vector = vectorizer.transform([preprocessed])
24prediction = model.predict(text_vector)[0]
25sentiment_result = "positive" if prediction == 1 else "negative"
26print(f"Sentiment: {sentiment_result}")sentiment_model.pkl: Trained Logistic Regression classifiersentiment_vectorizer.pkl: TF-IDF vectorizer with vocabulary