Spaces:
Build error
Build error
Keane Moraes
commited on
Commit
·
d3aa9b9
1
Parent(s):
4b1a525
nltk downloader
Browse files
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():
|