From 4eb164c31bc75cfc77d82b6327948b17c556ecb8 Mon Sep 17 00:00:00 2001 From: Charlene Leong Date: Thu, 24 Feb 2022 05:16:18 +0000 Subject: [PATCH] Fixing nltk deps --- zeroshot_topics/zeroshot_tm.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/zeroshot_topics/zeroshot_tm.py b/zeroshot_topics/zeroshot_tm.py index cd84cc0..7555c6c 100644 --- a/zeroshot_topics/zeroshot_tm.py +++ b/zeroshot_topics/zeroshot_tm.py @@ -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()