Update Dockerfile

Dockerfile

@@ -2,30 +2,21 @@ FROM python:3.9-slim
 
 WORKDIR /app
 
-# 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
+# Install dependencies
 RUN apt-get update && \
-    apt-get install -y --no-install-recommends gcc python3-dev libgomp1 && \
+    apt-get install -y --no-install-recommends gcc && \
     rm -rf /var/lib/apt/lists/*
 
-# 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}
+RUN pip install --no-cache-dir -r requirements.txt
 
 # Copy application code
 COPY app.py text2generation.py ./
 
 EXPOSE 7860
 
-# Health check endpoint
-HEALTHCHECK --interval=30s --timeout=30s --start-period=30s --retries=3 \
+# Health check
+HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
     CMD curl -f http://localhost:7860/health || exit 1
 
-# Run with Flask directly (no Gunicorn issues)
 CMD ["python", "app.py"]