diff --git a/Dockerfile b/Dockerfile index 99efd70..e3a3410 100644 --- a/Dockerfile +++ b/Dockerfile @@ -29,4 +29,4 @@ RUN python download_models.py EXPOSE 5000 # Define the command to run the server with parameters -CMD ["conda", "run", "-n", "xmen", "python3", "run_snomed_german_recommender.py", "--no-gpu", "--port", "5000", "index", "--num_recs", "10"] +CMD ["conda", "run", "--no-capture-output", "-n", "xmen", "python3", "-u", "run_snomed_german_recommender.py", "--no-gpu", "--port", "5000", "index", "--num_recs", "10"] diff --git a/run_snomed_german_recommender.py b/run_snomed_german_recommender.py index c933c00..40bd0f6 100644 --- a/run_snomed_german_recommender.py +++ b/run_snomed_german_recommender.py @@ -54,10 +54,10 @@ def run(): # Suppress InconsistentVersionWarning from TF-IDF vectorizer warnings.filterwarnings("ignore", category=InconsistentVersionWarning) - print("Loading xMEN SNOMED Linker...", flush=True) + print("Loading xMEN SNOMED Linker...") linker = default_ensemble(args.index_base_path, cuda=args.gpu) - print("Starting Ariadne server...", flush=True) + print("Starting Ariadne server...") server = Server() server.add_classifier("xmen_snomed", xMENSNOMEDLinker(linker, top_k=args.num_recs))