alihan/zeroshot_topics-nlp
1
0
Fork 0
mirror of https://github.com/maxent-ai/zeroshot_topics.git synced 2022-04-17 03:56:55 +03:00
Code Issues Projects Releases Wiki Activity

Merge pull request #6 from charyeezy/fix/missing_nltk_wordnet_deps

Browse Source 
Fixing nltk deps
This commit is contained in:
Rita Anjana
2022-02-25 13:54:19 +05:30
committed by GitHub
parent c77b12fcd8 4eb164c31b
commit f9a58ae5d1
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
zeroshot_topics/zeroshot_tm.py
View File

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