smhs16 commited on
Commit
9d2b2cb
·
verified ·
1 Parent(s): a4dc491

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -5,10 +5,15 @@ import nltk
5
  from nltk.corpus import stopwords
6
  from nltk.tokenize import word_tokenize
7
 
8
- # Ensure NLTK resources are downloaded
9
  nltk.download('punkt', quiet=True)
10
  nltk.download('stopwords', quiet=True)
11
 
 
 
 
 
 
 
12
  # Load model and vectorizer
13
  model = joblib.load("sentiment_model.pkl")
14
  vectorizer = joblib.load("tfidf_vectorizer.pkl")
 
5
  from nltk.corpus import stopwords
6
  from nltk.tokenize import word_tokenize
7
 
 
8
  nltk.download('punkt', quiet=True)
9
  nltk.download('stopwords', quiet=True)
10
 
11
+ from nltk.tokenize.punkt import PunktSentenceTokenizer, PunktTrainer
12
+ trainer = PunktTrainer()
13
+ tokenizer = PunktSentenceTokenizer(trainer.train(""))
14
+
15
+
16
+
17
  # Load model and vectorizer
18
  model = joblib.load("sentiment_model.pkl")
19
  vectorizer = joblib.load("tfidf_vectorizer.pkl")