undo detach

QubitPi · Oct 14, 2023 · b7992c2 · b7992c2
1 parent 4190b92
commit b7992c2
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git a/hashicorp/scripts/aws-base-tf-init.sh b/hashicorp/scripts/aws-base-tf-init.sh
@@ -6,7 +6,7 @@ touch /home/ubuntu/aws-base-tf-init.log
 export AWS_BASE_TF_INIT_LOG=/home/ubuntu/aws-base-tf-init.log
 echo "aws-base-tf-init started executing..."       >>$AWS_BASE_TF_INIT_LOG 2>&1
 
-sudo docker run --rm \
+sudo docker run --detach --rm \
   --memory=4000m \
   -p 5001:8080 \
   -v /home/ubuntu/theresa/mlflow_models/models/HanLPner:/opt/ml/model \

diff --git a/mlflow_models/HanLPner/HanLPner.py b/mlflow_models/HanLPner/HanLPner.py
@@ -1,8 +1,6 @@
 import hanlp
 import mlflow.pyfunc
 import pandas
-import torch
-import transformers
 
 
 class HanLPner(mlflow.pyfunc.PythonModel):
@@ -34,8 +32,6 @@ def predict(self, context, model_input):
                     'mlflow-skinny',
                     'mlflow[extras]',
                     'pandas=={}'.format(pandas.__version__),
-                    'torch=={}'.format(torch.__version__),
-                    'transformers=={}'.format(transformers.__version__),
                     'hanlp[amr, fasttext, full, tf]'
                 ],
             },