Arslan17121 commited on
Commit
7a2cbbf
·
verified ·
1 Parent(s): e431b19

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -4,8 +4,12 @@ import re
4
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
5
  from gtts import gTTS
6
  from sklearn.feature_extraction.text import CountVectorizer
 
7
  from nltk.sentiment import SentimentIntensityAnalyzer
8
 
 
 
 
9
  # Initialize necessary components
10
  tokenizer = AutoTokenizer.from_pretrained("google/pegasus-xsum")
11
  model = AutoModelForSeq2SeqLM.from_pretrained("google/pegasus-xsum")
 
4
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
5
  from gtts import gTTS
6
  from sklearn.feature_extraction.text import CountVectorizer
7
+ import nltk
8
  from nltk.sentiment import SentimentIntensityAnalyzer
9
 
10
+ # Download necessary NLTK data
11
+ nltk.download('vader_lexicon')
12
+
13
  # Initialize necessary components
14
  tokenizer = AutoTokenizer.from_pretrained("google/pegasus-xsum")
15
  model = AutoModelForSeq2SeqLM.from_pretrained("google/pegasus-xsum")