Added docker-file and edited the instructions for it

Dockerfile

@@ -0,0 +1,35 @@
+# Base image
+FROM ubuntu:20.04
+
+# Set working directory
+WORKDIR /app
+
+# Update and install required packages
+RUN apt-get update && \
+    apt-get install -y git-lfs wget && \
+    rm -rf /var/lib/apt/lists/*
+
+# Download and install Miniconda
+RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh && \
+    bash Miniconda3-latest-Linux-x86_64.sh -b -p /opt/conda && \
+    rm Miniconda3-latest-Linux-x86_64.sh
+
+# Set conda to automatically activate base environment on login
+RUN echo ". /opt/conda/etc/profile.d/conda.sh" >> ~/.bashrc && \
+    echo "conda activate base" >> ~/.bashrc
+
+# Create OpenChatKit environment
+COPY environment.yml .
+RUN conda env create -f environment.yml
+
+# Install Git LFS
+RUN git lfs install
+
+# Copy OpenChatKit code
+COPY . .
+
+# Prepare GPT-NeoX-20B model
+RUN python pretrained/GPT-NeoX-20B/prepare.py
+
+# Set entrypoint to bash shell
+ENTRYPOINT ["/bin/bash"]

README.md

@@ -43,6 +43,23 @@ This repo also uses [Git LFS](https://git-lfs.com/) to manage some files. Instal
 ```shell
 git lfs install
 ```
+### If you want to run the pacakge using Docker:
+
+Build the Docker image using the following command:
+```shell
+docker build -t openchatkit .
+```
+Run the Docker container using the following command:
+```shell
+docker run -it openchatkit
+```
+This will start a new bash shell in the container.
+Activate the OpenChatKit environment by running the following command:
+```shell
+conda activate OpenChatKit
+```
+
+
 
 # Pre-trained Weights