diff --git a/finetune/Dockerfile.ngc b/finetune/Dockerfile.ngc
index 3eb4e6c..4fb183b 100644
--- a/finetune/Dockerfile.ngc
+++ b/finetune/Dockerfile.ngc
@@ -79,7 +79,12 @@ EOT
 RUN <<EOT
 #!/bin/bash
 pip install -v -U git+https://github.com/facebookresearch/xformers.git@main#egg=xformers
+EOT
+
+RUN <<EOT
+#!/bin/bash
 pip install git+https://github.com/huggingface/transformers
+python -c "from transformers import pipeline; print(pipeline('sentiment-analysis')('I love you'))"
 EOT
 
 RUN <<EOT