Update Dockerfile

Dockerfile

@@ -1,21 +1,28 @@
 # Use an official Python runtime as a parent image
-FROM python:3.9-slim
+FROM python:3.10-slim
 
 # Set the working directory in the container
-WORKDIR /app
+WORKDIR /home/user/app
 
-# Copy the current directory contents into the container at /app
-COPY . /app
+# Copy the current directory contents into the container at /home/user/app
+COPY . /home/user/app
+
+# Install necessary system dependencies
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    libgl1-mesa-glx \
+    && rm -rf /var/lib/apt/lists/*
 
 # Install any needed packages specified in requirements.txt
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Make port 7860 available to the world outside this container
-EXPOSE 7860
-
-# Define environment variable
+# Set the environment variable for Hugging Face access token
 ARG ACC_TOKEN
 ENV AccToken=${ACC_TOKEN}
 
-# Run app.py when the container launches
-CMD ["python", "app.py"]
+# Expose the port the app runs on
+EXPOSE 7860
+
+# Define the command to run the application
+CMD ["python", "app.py"]
+RUN docker run -p 7860:7860 --env ACC_TOKEN=your_huggingface_token --cpus="32" --memory="200g" --gpus '"device=0,1,2,3,4,5,6,7,8,9"' your-image-name