Update Dockerfile

Dockerfile

@@ -23,7 +23,12 @@ RUN pip install --no-cache-dir \
     --find-links https://download.pytorch.org/whl/cpu \
     --extra-index-url https://pypi.org/simple && \
 # Install remaining packages that might have been skipped
-   
+    pip install --no-cache-dir \
+    accelerate \
+    transformers==4.36.2 \
+    timm==0.9.12 \
+    einops==0.7.0
+
 # Copy application code
 COPY . .