update base image

Dockerfile

@@ -1,12 +1,15 @@
 # Use an official PyTorch image as a parent image
-FROM pytorch/pytorch:1.9.0-cuda11.1-cudnn8-runtime
+FROM pytorch/pytorch:2.2.1-cuda12.1-cudnn8-devel
 
 # Set the working directory in the container
 WORKDIR /usr/src/app
 
 # Copy the current directory contents into the container at /usr/src/app
 COPY . .
 
+# Upgrade pip
+RUN python -m pip install --upgrade pip
+
 # Install any needed packages specified in requirements.txt
 RUN pip install --no-cache-dir -r requirements.txt
 
@@ -16,6 +19,5 @@ EXPOSE 80
 # Define environment variable
 ENV NAME World
 
-# Run cli_demo_sat.py when the container launches
-# Note: This command may need to be updated based on the specific entry point for the project.
-CMD ["python", "cli_demo_sat.py", "--from_pretrained", "cogvlm-chat", "--version", "chat_old", "--bf16", "--stream_chat"]
+# Run composite_demo/main.py when the container launches
+CMD ["python", "composite_demo/main.py"]