more logging

assets/main.py

@@ -14,11 +14,14 @@ import torch
 
 # multi-modal Model
 #   accepts both text and image content (or audio, etc.)
+print "loading model ..."
 model = BlipForImageTextRetrieval.from_pretrained("Salesforce/blip-itm-base-coco")
 processor = AutoProcessor.from_pretrained("Salesforce/blip-itm-base-coco")
 
+print "loading image ..."
 raw_image =  Image.open('./assets/pot-o-gold-my-little-pony-Derpy.jpeg').convert('RGB')
 
+print "processing ..."
 statements = [
     "an image of a horse",
     "a horse and a rainbow",

assets/script/entrypoint.sh

@@ -3,9 +3,12 @@ ROOT=$(cd `dirname $0` && pwd)
 
 set -e
 
+
+echo "initializing anaconda environment ..."
 eval "$(conda shell.bash hook)"
 conda init
 conda activate my-first-docker
 
 
+echo "executing main.py ..."
 python $ROOT/../main.py