docker file

Dockerfile

@@ -0,0 +1,39 @@
+# Use Python 3.9 slim image
+FROM python:3.9-slim
+
+# Set working directory
+WORKDIR /app
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    curl \
+    software-properties-common \
+    git \
+    && rm -rf /var/lib/apt/lists/*
+
+# Copy requirements first for better caching
+COPY requirements.txt .
+
+# Install Python dependencies
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy model files and application code
+COPY final-model/ ./final-model/
+COPY app.py .
+COPY model_utils.py .
+COPY index.html .
+
+# Create a non-root user
+RUN useradd -m -u 1000 appuser && chown -R appuser:appuser /app
+USER appuser
+
+# Expose port
+EXPOSE 8000
+
+# Health check
+HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
+    CMD curl -f http://localhost:8000/health || exit 1
+
+# Run the application
+CMD ["python", "app.py"] 

requirements.txt

@@ -1,8 +1,11 @@
-fastapi==0.104.1
-uvicorn[standard]==0.24.0
-torch>=2.0.0
-transformers>=4.35.0
-peft>=0.6.0
-accelerate>=0.24.0
-pydantic>=2.0.0
-python-multipart>=0.0.6 
+torch>=1.12.0
+transformers>=4.21.0
+peft>=0.3.0
+datasets>=2.0.0
+fastapi>=0.68.0
+uvicorn>=0.15.0
+pydantic>=1.8.0
+numpy>=1.21.0
+scikit-learn>=1.0.0
+accelerate>=0.12.0
+tokenizers>=0.12.0