Fixing nltk deps

This commit is contained in:
Charlene Leong
2022-02-24 05:16:18 +00:00
parent c77b12fcd8
commit 4eb164c31b

View File

@@ -1,6 +1,10 @@
import attr
from keybert import KeyBERT
from .utils import load_zeroshot_model
import nltk
nltk.download('wordnet')
from nltk.corpus import wordnet as wn
classifier = load_zeroshot_model()