Spaces:

atticus
/

image-text-retrival-huster

Runtime error

atticus commited on Mar 10, 2022

Commit

362a148

•

1 Parent(s): f733bb1

nltk

Files changed (1) hide show

app.py CHANGED Viewed

@@ -93,6 +93,8 @@ def search(mode, text):
     return res
 if __name__ == "__main__":
     # print("Loading model from:", model_path)
     checkpoint = torch.load(model_path, map_location=lambda storage, loc: storage)

     return res
 if __name__ == "__main__":
+    import nltk
+    nltk.download('punkt')
     # print("Loading model from:", model_path)
     checkpoint = torch.load(model_path, map_location=lambda storage, loc: storage)