From ca85c6d23d7441437f3cea6d3d621abc19c46c83 Mon Sep 17 00:00:00 2001 From: Michele Merler Date: Tue, 30 May 2023 15:59:49 -0400 Subject: [PATCH] Updated Dockerfile for Siamese (#209) * changed standardize output to be json compatible Signed-off-by: mmerler * allow input fields containerize and clustering to be either strings (from API) or dict (from command line) Signed-off-by: mmerler * fixed threshold reading from .ini config files in run_models Signed-off-by: mmerler * updated dev base image, added batching option for siamese model inference, set siamese model and tokenizer to local, updated github actions versions Signed-off-by: mmerler * updated Dockerfile for Siamese default Signed-off-by: mmerler --------- Signed-off-by: mmerler --- Dockerfile | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index ab0ac01..bb00d1d 100644 --- a/Dockerfile +++ b/Dockerfile @@ -34,7 +34,11 @@ RUN python -m pip install --upgrade pip wheel build setuptools; \ cd entity_standardizer; python -m build; pip install dist/entity_standardizer_tca-1.0-py3-none-any.whl; cd ..; \ pip install -r /app/requirements.txt; \ python benchmarks/generate_data.py; \ - python benchmarks/run_models.py; + mkdir -p models/deploy; \ + wget https://www.dropbox.com/s/bobcey7ufklw7mr/siamese_model.zip -O "./models/deploy/siamese_model.zip"; \ + unzip models/deploy/siamese_model.zip -d models/deploy/; \ + rm models/deploy/siamese_model.zip; \ + python benchmarks/run_models.py -model_type siamese -batch_size 5; RUN chown -R 1001:0 ./