Merge branch 'main' into ashahba/visualqna-llava-tgi

opea-project · Feb 1, 2025 · 27fe633 · 27fe633
2 parents 9905244 + 32d4f71
commit 27fe633
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/DocSum/ui/gradio/docsum_ui_gradio.py b/DocSum/ui/gradio/docsum_ui_gradio.py
@@ -313,11 +313,17 @@ def render(self):
 if __name__ == "__main__":
     import argparse
 
+    import nltk
+
     parser = argparse.ArgumentParser()
     parser.add_argument("--host", type=str, default="0.0.0.0")
     parser.add_argument("--port", type=int, default=5173)
 
     args = parser.parse_args()
     logger.info(">>> Starting server at %s:%d", args.host, args.port)
 
+    # Needed for UnstructuredURLLoader when reading content from a URL
+    nltk.download("punkt_tab")
+    nltk.download("averaged_perceptron_tagger_eng")
+
     uvicorn.run(app, host=args.host, port=args.port)