nltk fixed

jiatastic · Jul 24, 2023 · bc6364f · bc6364f
1 parent e146c59
commit bc6364f
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 0 deletions.
diff --git a/pages/Behavioral Screen.py b/pages/Behavioral Screen.py
@@ -20,6 +20,7 @@
 from langchain.vectorstores import FAISS
 import base64
 from IPython.display import Audio
+import nltk
 
 ### ————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————
 def load_lottiefile(filepath: str):
@@ -53,6 +54,7 @@ def update_markdown(audio_md):
     update_markdown(global_audio_md)
 
 def save_vector(text: str):
+    nltk.download('punkt')
     text_splitter = NLTKTextSplitter()
     texts = text_splitter.split_text(text)
     # Create emebeddings

diff --git a/pages/Professional Screen.py b/pages/Professional Screen.py
@@ -38,6 +38,8 @@ class Message:
 
 def save_vector(text):
     """embeddings"""
+
+    nltk.download('punkt')
     text_splitter = NLTKTextSplitter()
     texts = text_splitter.split_text(text)
      # Create emebeddings

diff --git a/pages/Resume Screen.py b/pages/Resume Screen.py
@@ -19,6 +19,7 @@
 from streamlit_lottie import st_lottie
 import json
 from IPython.display import Audio
+import nltk
 
 ### ——————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————————
 def load_lottiefile(filepath: str):
@@ -37,6 +38,8 @@ class Message:
     message: str
 
 def save_vector(resume):
+    """embeddings"""
+    nltk.download('punkt')
     pdf_reader = PdfReader(resume)
     text = ""
     for page in pdf_reader.pages: