Update Dockerfile

Dockerfile

@@ -1,42 +1,29 @@
-# Use official Python 3.12 base image
-FROM python:3.12-slim
+# Base image with CUDA and PyTorch
+FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime
 
-# Install system dependencies
-RUN apt-get update && \
-    apt-get install -y --no-install-recommends \
-      git curl libgl1 libglib2.0-0 && \
-    rm -rf /var/lib/apt/lists/*
+# Install dependencies
+RUN apt-get update && apt-get install -y \
+    git \
+    curl \
+    wget \
+    libgl1 \
+    libglib2.0-0 \
+    && rm -rf /var/lib/apt/lists/*
 
-# Create a non-root user
-RUN useradd -m -u 1000 user
+# Create app directory
+WORKDIR /app
 
-# Switch to non-root user before copying and installing
-USER user
+# Copy requirements
+COPY requirements.txt .
 
-# Set up home and cache environment variables
-ENV HOME=/home/user \
-    PATH=/home/user/.local/bin:$PATH \
-    HF_HOME=/home/user/.cache/huggingface \
-    TRANSFORMERS_CACHE=/home/user/.cache/huggingface/transformers \
-    HF_DATASETS_CACHE=/home/user/.cache/huggingface/datasets
+# Install Python dependencies
+RUN pip install --upgrade pip && pip install -r requirements.txt
 
-# Make sure cache directories are writable
-RUN mkdir -p $HF_HOME/hub $TRANSFORMERS_CACHE $HF_DATASETS_CACHE && \
-    chmod -R 777 /home/user/.cache
+# Copy the rest of the app
+COPY . .
 
-# Set working directory
-WORKDIR $HOME/app
-
-# Copy dependencies and install
-COPY --chown=user:user requirements.txt .
-RUN pip install --upgrade pip && \
-    pip install --no-cache-dir -r requirements.txt
-
-# Copy application code
-COPY --chown=user:user . .
-
-# Expose the default port
+# Expose port if running a web interface
 EXPOSE 7860
 
-# Run the FastAPI app
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# Entry point
+CMD ["python", "app.py"]