ahm14 commited on
Commit
d89fbaa
·
verified ·
1 Parent(s): 1ff45cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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