Update Dockerfile

Dockerfile

@@ -2,7 +2,7 @@ FROM python:3.9-slim
 
 WORKDIR /app
 
-# Install system dependencies (for transformers and other requirements)
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
     git \
     gcc \
@@ -10,10 +10,10 @@ RUN apt-get update && apt-get install -y \
     make \
     && rm -rf /var/lib/apt/lists/*
 
-# Create cache directory with proper permissions
+# Create cache directory
 RUN mkdir -p /.cache && chmod -R 777 /.cache
 
-# Set environment variables for Hugging Face cache
+# Set environment variables
 ENV TRANSFORMERS_CACHE=/.cache
 ENV HF_HOME=/.cache
 ENV HUGGINGFACE_HUB_CACHE=/.cache
@@ -22,12 +22,8 @@ ENV HUGGINGFACE_HUB_CACHE=/.cache
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Optionally pre-download the tokenizer during build
-# Uncomment if you want to download during build (for smaller models)
-# RUN python -c "from transformers import AutoTokenizer; AutoTokenizer.from_pretrained('codellama/CodeLlama-7b-hf')"
-
-# Copy code
-COPY . .
+# Copy application code
+COPY main.py .
 
 # Expose port
 EXPOSE 7860