Install gradio in this image, instead of base DiffDock-Pocker image.

Dockerfile

@@ -1,13 +1,16 @@
-FROM silterra/diffdock-pocket-dev
+# This container is a basic web frontend, we do the heavy-lifting in the
+# main DiffDock-Pocket image.
+FROM silterra/diffdock-pocket
 
 USER $APPUSER
 WORKDIR $HOME/app
 
-COPY --chown=$APPUSER: . $HOME/app
+COPY --chown=$APPUSER:$APPUSER . $HOME/app
+
+RUN ~/bin/micromamba install -n $ENV_NAME -c conda-forge -c defaults --file requirements.txt
 
 # Expose port for web service
-ENV PORT=7860
-EXPOSE $PORT
+EXPOSE 7860
 
 # Run app under micromamba environment
 CMD ["sh", "-c", "micromamba run -n ${ENV_NAME} python3 main.py"]

basic_environment.yml → environment.yml

@@ -9,4 +9,4 @@ dependencies:
   - pip=23.3.1
   - pytorch=1.13.1
   - pip:
-    - gradio==3.50
+    - -r requirements.txt

requirements.txt

@@ -1,4 +1,2 @@
-gradio
-torch
-torchvision
-requests
+gradio==3.50
+requests==2.31.0