add library

app.py

@@ -12,8 +12,10 @@ import matplotlib.pyplot as plt
 import models
 import string
 import nltk
-nltk.download('punkt')
-nltk.download('averaged_perceptron_tagger')
+# nltk.download('punkt')
+# nltk.download('averaged_perceptron_tagger')
+
+from nltk.tokenize import word_tokenize
 import torchvision
 
 args = default_argument_parser().parse_args()
@@ -38,7 +40,7 @@ def run(sketch, caption, threshold, seed):
     caption = caption.replace('\n',' ')
     translator = str.maketrans('', '', string.punctuation)
     caption = caption.translate(translator).lower()
-    words = nltk.word_tokenize(caption)
+    words = word_tokenize(caption)
     classes = get_noun_phrase(words)
     if len(classes) ==0:
         classes = [caption]