Try to fix import error issue

Dockerfile

@@ -8,7 +8,14 @@ RUN pip install --upgrade pip setuptools wheel
 RUN pip install cmake
 
 # Install system build dependencies
-RUN apt-get clean && apt-get -y update && apt-get install -y build-essential cmake libopenblas-dev liblapack-dev libopenblas-dev liblapack-dev
+RUN apt-get clean && apt-get -y update && \
+    apt-get install -y --no-install-recommends \
+        build-essential \
+        cmake \
+        libopenblas-dev \
+        liblapack-dev \
+        libx11-dev && \
+    rm -rf /var/lib/apt/lists/*
 
 ENV CMAKE_BUILD_PARALLEL_LEVEL=4
 
@@ -26,7 +33,14 @@ FROM python:3.12-slim AS runner
 WORKDIR /app
 
 # Install runtime dependency: libopenblas.so.0 is provided by libopenblas-base.
-RUN apt-get update && apt-get install -y build-essential cmake libopenblas-dev liblapack-dev libopenblas-dev liblapack-dev
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+        build-essential \
+        cmake \
+        libopenblas-dev \
+        liblapack-dev \
+        libx11-6 && \
+    rm -rf /var/lib/apt/lists/*
 
 ENV HF_HOME=/tmp/huggingface
 ENV HUGGINGFACE_HUB_CACHE=/tmp/huggingface
@@ -39,7 +53,7 @@ RUN mkdir -p /tmp/model /tmp/huggingface /tmp/.cache /tmp/.gradio /tmp/.gradio/c
 
 COPY --from=builder /app/venv venv
 
-COPY app.py models.py test_functions.py s3.py ./
+COPY app.py models.py test_functions.py test_compress.py ./
 COPY examples/ /app/examples/
 COPY assets/ /app/assets/
 

app.py

@@ -12,7 +12,7 @@ from dotenv import load_dotenv
 
 load_dotenv()
 
-from s3 import build_context, imagine
+from test_compress import build_context, imagine
 
 # Model download & caching directory (created in Dockerfile)
 MODEL_DIR = "/tmp/model"