update Dockerfile and requirements

Dockerfile

@@ -1,4 +1,4 @@
-FROM python:3.10 AS build
+FROM python@sha256:e2c7fb05741c735679b26eda7dd34575151079f8c615875fbefe401972b14d85 AS build
 
 RUN useradd -m -u 1000 user
 USER user
@@ -18,7 +18,12 @@ RUN pip install --no-cache-dir .[cpu,gui] --extra-index-url https://download.pyt
 
 FROM build AS final
 
-COPY --chown=user . .
+# copy dirs used at inference time
+COPY --chown=user configs configs
+COPY --chown=user shimnet shimnet
+# copy files used at inference time
+COPY --chown=user predict-gui.py .
+COPY --chown=user download_files.py .
 
 # download weights
 RUN python download_files.py --overwrite

pyproject.toml

@@ -48,7 +48,8 @@ gui = [
     "plotly==6.0.1",
     "ipykernel",
     "psutil",
-    "pexpect"
+    "pexpect",
+    "requests"
 ]
 
 # lightweight verion for Colab or another environment with pre-installed torch and common packages