update with hf cache

Dockerfile

@@ -4,6 +4,9 @@ FROM python:3.9-slim
 # Set the working directory in the container
 WORKDIR /app
 
+# Create a directory for the Hugging Face cache
+RUN mkdir -p /app/hf_cache
+
 # Copy the current directory contents into the container at /app
 COPY . /app
 
@@ -13,7 +16,8 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Make port 8000 available to the world outside this container
 EXPOSE 8000
 
-# Define environment variable
+# Set environment variable for the Hugging Face cache
+ENV TRANSFORMERS_CACHE=/app/hf_cache
 ENV NAME World
 
 # Run app.py when the container launches

depth_est.py

@@ -4,6 +4,10 @@ import base64
 import requests
 import io
 import numpy as np
+import os
+
+
+os.environ['TRANSFORMERS_CACHE'] = '/app/hf_cache'
 
 def convert_img_base64(img):
     with open(img, 'rb') as file: