Update Dockerfile

Dockerfile

@@ -7,6 +7,17 @@ RUN useradd -m -u 1000 user
 USER user
 ENV PATH="/home/user/.local/bin:$PATH"
 
+
+RUN mkdir -p /cache/huggingface /cache/torch \
+    && python -c "from transformers import AutoModelForCausalLM, AutoTokenizer; \
+                  AutoModelForCausalLM.from_pretrained('microsoft/phi-2'); \
+                  AutoTokenizer.from_pretrained('microsoft/phi-2')" \
+    && python -c "from torchvision.models.detection import fasterrcnn_resnet50_fpn, FasterRCNN_ResNet50_FPN_Weights; \
+                  weights = FasterRCNN_ResNet50_FPN_Weights.DEFAULT; \
+                  model = fasterrcnn_resnet50_fpn(weights=weights)"
+
+
+
 WORKDIR /app
 
 COPY --chown=user ./requirements.txt requirements.txt