Update Dockerfile

Dockerfile

@@ -1,6 +1,9 @@
 FROM python:3.9-slim
 
-# Set working directory
+# Create a non-root user
+RUN useradd -m appuser
+
+# Set the working directory
 WORKDIR /code
 
 # Install system dependencies
@@ -10,25 +13,27 @@ RUN apt-get update && apt-get install -y \
     curl \
     && rm -rf /var/lib/apt/lists/*
 
-
-RUN mkdir -p /code/models/.cache/huggingface
-
-# Copy requirements and install
-COPY app/requirements.txt .
+# Copy requirements file and install Python dependencies
+COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy the entire app folder
-COPY app/ ./app/
+# Copy the application code
+COPY app /code/app
+COPY main.py .
+
+# Create models and cache directories and set correct permissions
+RUN mkdir -p /code/models/.cache/huggingface && \
+    chown -R appuser:appuser /code/models
 
-# Set the working directory to app/
-WORKDIR /code/app
+# Set environment variables
+ENV HOST=0.0.0.0
+ENV PORT=7860
 
 # Expose the port
 EXPOSE 7860
 
-# Environment variables (optional)
-ENV HOST=0.0.0.0
-ENV PORT=7860
+# Switch to the non-root user
+USER appuser
 
-# Run the FastAPI app
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+# Command to run the FastAPI app
+CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]