Update Dockerfile

Dockerfile

@@ -1,22 +1,34 @@
-# Use the official Python image as a base image
-FROM python:3.9
+# Use Python 3.10 for compatibility with newer packages
+FROM python:3.10
 
-# Set the working directory in the container
-WORKDIR /app
+# Install required system dependencies
+RUN apt-get update && apt-get install -y \
+    libgl1-mesa-glx \
+    libglib2.0-0
 
-# Copy the requirements file to the container
-COPY requirements.txt .
+RUN useradd user
 
-# Install any necessary dependencies
-RUN pip install --no-cache-dir -r requirements.txt
+USER user
 
-# Copy the rest of the application code to the container
-COPY . .
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH
 
-# Expose the port that the FastAPI app will run on
-EXPOSE 8000
+# Set up a working directory
+WORKDIR $HOME/app
 
-##
+COPY --chown=user ./ $HOME/app
 
-# Command to run the FastAPI app using Uvicorn
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8000"]
+# Install dependencies from the requirements file
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+# Install PyTorch and torchvision
+RUN pip install torch torchvision
+
+# Set environment variable to indicate PyTorch usage
+ENV USE_TORCH=1 
+
+# Expose port 7860 for the app
+EXPOSE 1326
+
+# Command to run FastAPI app with Uvicorn
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "1326"]