Upload 2 files

Dockerfile

@@ -0,0 +1,38 @@
+# Use an official Python runtime as a parent image
+FROM python:3.9-slim
+
+# Set working directory
+WORKDIR /app
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/*
+
+# Create a non-root user
+RUN useradd -m -u 1000 appuser
+
+# Copy requirements first to leverage Docker cache
+COPY requirements.txt .
+
+# Install Python dependencies
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Create necessary directories and set permissions
+RUN mkdir -p /app/saved_models /app/tokenizer /app/predictions /app/.cache \
+    && chown -R appuser:appuser /app
+
+# Switch to non-root user
+USER appuser
+
+# Copy the application code
+COPY --chown=appuser:appuser . .
+
+# Download the BERT tokenizer
+RUN python -c "from transformers import BertTokenizer; BertTokenizer.from_pretrained('bert-base-uncased', cache_dir='/app/.cache')"
+
+# Expose the port the app runs on (7860 for Hugging Face Spaces)
+EXPOSE 7860
+
+# Command to run the application
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"] 

docker-compose.yml

@@ -0,0 +1,15 @@
+version: '3.8'
+
+services:
+  bert-api:
+    build: .
+    ports:
+      - "8000:8000"
+    volumes:
+      - ../saved_models:/app/saved_models
+      - ../tokenizer:/app/tokenizer
+      - ../predictions:/app/predictions
+      - ../label_encoders.pkl:/app/label_encoders.pkl
+    environment:
+      - PYTHONUNBUFFERED=1
+    restart: unless-stopped