Update Dockerfile

Dockerfile

@@ -1,40 +1,31 @@
-# Stage 1: Builder
-FROM python:3.9-slim as builder
-
-WORKDIR /app
-COPY requirements.txt .
-
-RUN apt-get update && \
-    apt-get install -y --no-install-recommends gcc python3-dev && \
-    pip install --no-cache-dir -r requirements.txt gunicorn && \
-    rm -rf /var/lib/apt/lists/*
-
-# Stage 2: Runtime
 FROM python:3.9-slim
 
 WORKDIR /app
 
-# Install system dependencies
+# Set cache directory and fix numpy version
+ENV TRANSFORMERS_CACHE=/tmp/huggingface-cache \
+    NUMBA_CACHE_DIR=/tmp/numba-cache \
+    HF_HOME=/tmp/huggingface-home
+
+# Install system dependencies with cleanup
 RUN apt-get update && \
-    apt-get install -y --no-install-recommends libgomp1 && \
+    apt-get install -y --no-install-recommends gcc python3-dev libgomp1 && \
     rm -rf /var/lib/apt/lists/*
 
-# Copy Python dependencies (including gunicorn)
-COPY --from=builder /usr/local/lib/python3.9/site-packages /usr/local/lib/python3.9/site-packages
-COPY --from=builder /usr/local/bin/gunicorn /usr/local/bin/
+# Install Python dependencies with pinned numpy version
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt numpy==1.26.4 && \
+    mkdir -p ${TRANSFORMERS_CACHE} && \
+    chmod 777 ${TRANSFORMERS_CACHE}
 
 # Copy application code
 COPY app.py text2generation.py ./
 
-# Expose Hugging Face's default port
 EXPOSE 7860
 
-# Health check
+# Health check endpoint
 HEALTHCHECK --interval=30s --timeout=30s --start-period=30s --retries=3 \
     CMD curl -f http://localhost:7860/health || exit 1
 
-# Verify gunicorn is available
-RUN which gunicorn && gunicorn --version
-
-# Run with Gunicorn
-CMD ["gunicorn", "--bind", "0.0.0.0:7860", "--workers", "1", "--timeout", "120", "app:app"]
+# Run with Flask directly (no Gunicorn issues)
+CMD ["python", "app.py"]