Update Dockerfile

Dockerfile

@@ -28,8 +28,11 @@ RUN mkdir build && \
 
 FROM ${BASE_CUDA_RUN_CONTAINER} as runtime
 RUN apt-get update --fix-missing && \
-    apt-get install -y --no-install-recommends wget libgomp1 && \
+    apt-get install -y --no-install-recommends wget libgomp1 python3 python3-pip && \
     rm -rf /var/lib/apt/lists/*
+    
+# Install huggingface_hub[cli]
+RUN pip3 install huggingface_hub[cli]
 
 WORKDIR /app