mirror of
https://github.com/maxent-ai/zeroshot_topics.git
synced 2022-04-17 03:56:55 +03:00
Fixing nltk deps
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user