now seems that's all here

Dockerfile

@@ -3,20 +3,29 @@ FROM pytorch/pytorch:latest
 # prevent apt from hanging
 ARG DEBIAN_FRONTEND=noninteractive
 
+# create workspace
 ENV HOME /workspace
-
 WORKDIR $HOME
 
 # dependency: lang-segment-anything
 RUN apt update
+
+# installing system dependencies:
 RUN apt install -y git
 RUN apt install libgl1-mesa-glx -y
 RUN apt install libglib2.0-0 -y
-RUN git clone https://github.com/kauevestena/lang-segment-anything.git
+
+# copy source code:
+COPY . $HOME/lang-segment-anything
+
+# installing python dependencies:
 WORKDIR $HOME/lang-segment-anything
 RUN pip install -e .
 
 # running the basic test,
-# then it will also be helding the weights inside the image,
+# then it will held the weights inside the image,
 # so no "cold start"
-RUN python running_test.py
+RUN python running_test.py
+
+# running the app:
+CMD ["lightning", "run", "app", "app.py"]

README.md

@@ -51,6 +51,7 @@ Build and run the image.
  docker run --gpus all -it lang-segment-anything:latest
  ```
 
+If you want a shared folder you can add a volume with `-v <host_source_dir>:<container_dest_dir>` example: `-v ./data:/workspace/data`
 
 
 ### Usage