Update Dockerfile

Dockerfile

@@ -1,40 +1,47 @@
-# Base image
-FROM python:3.9-slim-buster
+# Use the official Hugging Face Docker image as base
+FROM python:3.9-slim
 
-# Environment variables
+# Set environment variables
 ENV PYTHONDONTWRITEBYTECODE=1 \
     PYTHONUNBUFFERED=1 \
-    TOKENIZERS_PARALLELISM=false \
-    HF_HOME=/app/cache \
-    HUGGINGFACE_HUB_CACHE=/app/cache
+    HF_HOME=/cache \
+    HUGGINGFACE_HUB_CACHE=/cache \
+    PIP_NO_CACHE_DIR=1 \
+    PORT=8000
 
-# Set working directory
+# Create and set working directory
 WORKDIR /app
 
-# Install OS dependencies
+# Install system dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
     build-essential \
-    libgomp1 \
-    curl \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy files
-COPY . .
+# Copy requirements first for better caching
+COPY requirements.txt .
 
 # Install Python dependencies
 RUN pip install --upgrade pip && \
     pip install --no-cache-dir -r requirements.txt
 
-# Preload models into cache
-RUN python -c "\
-from sentence_transformers import SentenceTransformer; \
-SentenceTransformer('all-MiniLM-L6-v2', cache_folder='/app/cache'); \
-from transformers import AutoModelForCausalLM, AutoTokenizer; \
-AutoModelForCausalLM.from_pretrained('distilgpt2', cache_dir='/app/cache'); \
-AutoTokenizer.from_pretrained('distilgpt2', cache_dir='/app/cache')"
+# Copy the rest of the application
+COPY . .
 
-# Hugging Face Spaces expects app on port 7860
-ENV PORT=7860
+# Create necessary directories
+RUN mkdir -p /app/data /cache
 
-# Run with Gunicorn (bind to PORT)
-CMD ["gunicorn", "--bind", "0.0.0.0:7860", "app:app"]
+# Download models during build (optional)
+RUN python -c "\
+try: \
+    from sentence_transformers import SentenceTransformer; \
+    SentenceTransformer('all-MiniLM-L6-v2', cache_folder='/cache'); \
+    from transformers import AutoModelForCausalLM, AutoTokenizer; \
+    AutoModelForCausalLM.from_pretrained('distilgpt2', cache_dir='/cache'); \
+    AutoTokenizer.from_pretrained('distilgpt2', cache_dir='/cache'); \
+except Exception as e: \
+    print(f'Model preloading failed: {str(e)}'); \
+    raise \
+"
+
+# Run the application
+CMD gunicorn --bind 0.0.0.0:$PORT --workers 2 --timeout 120 --preload app:app