add checker for punkt

OpenAdaptAI · Jun 21, 2023 · 4bcb474 · 4bcb474
1 parent 0ed8dfd
commit 4bcb474
Showing 1 changed file with 7 additions and 8 deletions.
diff --git a/openadapt/strategies/mixins/summary.py b/openadapt/strategies/mixins/summary.py
@@ -46,14 +46,13 @@ def get_summary(
         text: str,
         num_sentences: int,
     ) -> str:
+        while(True):
+            try:
+                Tokenizer("english")
+                break
+            except:
+                nltk.download('punkt')
+
         parser = PlaintextParser.from_string(text, Tokenizer("english"))
         summarized = self.summarizer(parser.document, num_sentences)
         return summarized
-
-if __name__ == "__main__":
-    while(True):
-        try:
-            Tokenizer("english")
-            break
-        except:
-            nltk.download('punkt')