Update Dockerfile

Dockerfile

@@ -1,36 +1,15 @@
-# Build stage
-FROM python:3.10 AS builder
-WORKDIR /app
-
-# Set up cache directories
-RUN mkdir -p /tmp/huggingface_cache && chmod 777 /tmp/huggingface_cache
-ENV HF_HOME=/tmp/huggingface_cache
-ENV TRANSFORMERS_CACHE=/tmp/huggingface_cache
-
-# Copy requirements and install
-COPY requirements.txt .
-RUN pip install -r requirements.txt
-
-# Pre-download models
-COPY download_models.py .
-RUN python download_models.py
-
-# Runtime stage
 FROM python:3.10-slim
-WORKDIR /app
 
-# Set up cache directories in runtime stage
-RUN mkdir -p /tmp/huggingface_cache && chmod 777 /tmp/huggingface_cache
-ENV HF_HOME=/tmp/huggingface_cache
-ENV TRANSFORMERS_CACHE=/tmp/huggingface_cache
+WORKDIR /app
 
-# Copy everything from builder
-COPY --from=builder /usr/local/lib/python3.10 /usr/local/lib/python3.10
-COPY --from=builder /usr/local/bin /usr/local/bin
-COPY --from=builder /tmp/huggingface_cache /tmp/huggingface_cache
+# Install build dependencies (if any are needed by pip packages)
+# Remove build dependencies afterward if minimizing size is critical
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
 
 # Copy application code
 COPY . .
 
 EXPOSE 50010
+
 CMD ["python3", "app.py"]