Update app.py
Browse files
app.py
CHANGED
|
@@ -16,6 +16,7 @@ from langchain_groq import ChatGroq
|
|
| 16 |
from langchain_core.output_parsers import StrOutputParser
|
| 17 |
from langchain_core.prompts import ChatPromptTemplate
|
| 18 |
from transformers import pipeline
|
|
|
|
| 19 |
from rake_nltk import Rake
|
| 20 |
|
| 21 |
# Load environment variables
|
|
@@ -35,6 +36,7 @@ llm = ChatGroq(temperature=0.5, groq_api_key=GROQ_API_KEY, model_name="llama3-8b
|
|
| 35 |
|
| 36 |
# Download required NLTK resources
|
| 37 |
nltk.download("punkt")
|
|
|
|
| 38 |
nltk.download("stopwords")
|
| 39 |
|
| 40 |
# Tone categories for fallback method
|
|
|
|
| 16 |
from langchain_core.output_parsers import StrOutputParser
|
| 17 |
from langchain_core.prompts import ChatPromptTemplate
|
| 18 |
from transformers import pipeline
|
| 19 |
+
from nltk.tokenize import sent_tokenize
|
| 20 |
from rake_nltk import Rake
|
| 21 |
|
| 22 |
# Load environment variables
|
|
|
|
| 36 |
|
| 37 |
# Download required NLTK resources
|
| 38 |
nltk.download("punkt")
|
| 39 |
+
nltk.download("punkt_tab")
|
| 40 |
nltk.download("stopwords")
|
| 41 |
|
| 42 |
# Tone categories for fallback method
|