Spaces:
Running
Running
Update app.py
Browse files
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")
|