Initial commit

Dockerfile

@@ -1,43 +1,36 @@
-# Use an official Python runtime as a parent image
+# Dockerfile
+
 FROM python:3.9-slim
 
-# Install git and git-lfs
+# Install dependencies
 RUN apt-get update && apt-get install -y git git-lfs && git lfs install
 
-# Create a non-root user 'appuser'
+# Set environment variables
+ENV HF_HOME="/home/appuser/.cache/huggingface"
+ENV TRANSFORMERS_CACHE="/home/appuser/.cache/huggingface/transformers"
+ENV HF_DATASETS_CACHE="/home/appuser/.cache/huggingface/datasets"
+
+# Create a non-root user
 RUN useradd -ms /bin/bash appuser
 
 # Set the working directory
 WORKDIR /home/appuser/app
 
-# Copy requirements file
+# Copy requirements and install
 COPY requirements.txt .
-
-# Install required packages
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy application code
+# Copy the application code
 COPY . .
 
-# Set environment variables for cache directories
-ENV HF_HOME=/home/appuser/app/.cache
-ENV HF_DATASETS_CACHE=/home/appuser/app/.cache
-
-# Create the cache directory
-RUN mkdir -p /home/appuser/app/.cache
-
-# Change ownership of the application files
-RUN chown -R appuser:appuser /home/appuser/app
+# Change ownership
+RUN chown -R appuser:appuser /home/appuser
 
 # Switch to non-root user
 USER appuser
 
-# Pre-download models and datasets
-RUN python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('all-MiniLM-L6-v2')"
-RUN python -c "from datasets import load_dataset; load_dataset('Gustavosta/Stable-Diffusion-Prompts')"
-
-# Expose port 7860
+# Expose the port
 EXPOSE 7860
 
-# Command to run the API
+# Command to run the application
 CMD ["uvicorn", "api:app", "--host", "0.0.0.0", "--port", "7860"]