Update Dockerfile

Dockerfile

@@ -30,6 +30,7 @@ WORKDIR /app
 
 # Upgrade pip and install core dependencies first
 RUN python3 -m pip install --no-cache-dir --upgrade pip setuptools wheel packaging ninja
+RUN pip install --no-cache-dir "numpy<2"
 
 # Install PyTorch with CUDA 11.8 support
 RUN pip install --no-cache-dir \
@@ -38,11 +39,8 @@ RUN pip install --no-cache-dir \
     torchaudio==2.1.2 \
     --index-url https://download.pytorch.org/whl/cu118
 
-# Install flash-attn dependencies first
-RUN pip install --no-cache-dir \
-    einops==0.7.0 \
-    triton==2.1.0 \
-    && pip install --no-cache-dir flash-attn==2.3.3 --no-build-isolation
+# Install flash-attn with compatible version
+RUN pip install --no-cache-dir flash-attn==2.3.3 --no-build-isolation
 
 # Copy and install remaining requirements
 COPY requirements.txt .
@@ -57,4 +55,8 @@ RUN useradd -m -u 1000 user && \
 
 USER user
 
+# Set environment variable to suppress HuggingFace warnings
+ENV HF_HUB_DISABLE_PROGRESS=1
+ENV TF_CPP_MIN_LOG_LEVEL=3
+
 CMD ["uvicorn", "download_api:app", "--host", "0.0.0.0", "--port", "7860"]