Keane Moraes commited on
Commit
d3aa9b9
·
1 Parent(s): 4b1a525

nltk downloader

Browse files
Files changed (1) hide show
  1. utils.py +3 -0
utils.py CHANGED
@@ -1,6 +1,8 @@
1
  import streamlit as st
2
  from keybert import KeyBERT
 
3
  from nltk.corpus import stopwords
 
4
  from transformers import AutoTokenizer
5
  import os, re, json
6
  import openai
@@ -11,6 +13,7 @@ import numpy as np
11
  from sentence_transformers import SentenceTransformer
12
 
13
  MODEL = 'all-MiniLM-L6-v2'
 
14
 
15
  @st.cache_data
16
  def load_autotoken():
 
1
  import streamlit as st
2
  from keybert import KeyBERT
3
+ import nltk
4
  from nltk.corpus import stopwords
5
+
6
  from transformers import AutoTokenizer
7
  import os, re, json
8
  import openai
 
13
  from sentence_transformers import SentenceTransformer
14
 
15
  MODEL = 'all-MiniLM-L6-v2'
16
+ nltk.download('stopwords')
17
 
18
  @st.cache_data
19
  def load_autotoken():