update docker

Dockerfile

@@ -55,8 +55,9 @@ RUN conda install pillow -y
 
 # RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt
 
-RUN conda install -c pytorch pytorch==1.7.1 torchvision==0.8.2 
+# RUN conda install -c pytorch pytorch==1.7.1 torchvision==0.8.2 
 # RUN pip install torchvision-cpu==0.8.2
+RUN pip3 install torch==1.11.0+cu115 torchvision==0.12.0+cu115 --extra-index-url https://download.pytorch.org/whl/cu115
 
 # WORKDIR /home/user
 
@@ -64,7 +65,6 @@ RUN conda install -c pytorch pytorch==1.7.1 torchvision==0.8.2
 
 RUN pip3 install openmim
 
-
 RUN conda install -c conda-forge gradio -y 
 
  
@@ -75,18 +75,20 @@ RUN conda install -c conda-forge gradio -y
 WORKDIR /home/user
 
 RUN --mount=type=secret,id=git_token,mode=0444,required=true \
-    git clone https://$(cat /run/secrets/git_token)@github.com/NASA-IMPACT/hls-foundation-os.git
+    git clone https://$(cat /run/secrets/git_token)@github.com/NASA-IMPACT/hls-foundation-os.git@mmseg-only
 
 
 WORKDIR hls-foundation-os 
 
-RUN pip3 install fine-tuning-examples/
+RUN pip3 install -e .
 
 
 # RUN --mount=type=secret,id=git_token,mode=0444,required=true \
 #     pip3 install git+https://$(cat /run/secrets/git_token)@github.com/NASA-IMPACT/hls-foundation-os.git@mmseg-only
 
-RUN mim install mmcv-full==1.5.0
+# RUN mim install mmcv-full==1.5.0
+RUN mim install mmcv-full==1.6.2 -f https://download.openmmlab.com/mmcv/dist/11.5/1.11.0/index.html
+
 
 # Set the working directory to the user's home directory
 WORKDIR $HOME/app