Inherit from other container again

Dockerfile

@@ -1,33 +1,14 @@
-FROM nvidia/cuda:11.7.1-devel-ubuntu22.04
-
-RUN apt-get update -y
-RUN apt-get install -y sudo wget curl nano git \
-    python3 python3-pip && rm -rf /var/lib/apt/lists/*
-
-# Create a group and user
-ENV APPUSER="appuser"
-ENV HOME=/home/$APPUSER
-RUN useradd -m -u 1000 $APPUSER
+FROM silterra/diffdock-pocket-dev
 
 USER $APPUSER
-WORKDIR $HOME
-
-# Set home to the user's home directory
-ENV HOME=/home/$APPUSER \
-    PATH=/home/$APPUSER/.local/bin:$PATH
-
-COPY --chown=$APPUSER ./requirements.txt $HOME/app/requirements.txt
 WORKDIR $HOME/app
 
-RUN pip install --no-cache-dir --user -r requirements.txt
-
 COPY --chown=$APPUSER . $HOME/app
 
 # Expose port for web service
 ENV PORT=7860
 EXPOSE $PORT
 
-# Run streamlit app under conda environment
-# CMD ["sh", "-c", "streamlit run --server.port=$PORT --server.address=0.0.0.0 app.py"]
-
-CMD ["python3", "main.py"]
+# Run app under micromamba environment
+CMD ["sh", "-c", "micromamba run -n ${ENV_NAME} python3 main.py"]
+# CMD ["python3", "main.py"]

main.py

@@ -1,11 +1,12 @@
 import gradio as gr
 import torch
-import requests
-from torchvision import transforms
 
-model = torch.hub.load("pytorch/vision:v0.6.0", "resnet18", pretrained=True).eval()
-response = requests.get("https://git.io/JJkYN")
-labels = response.text.split("\n")
+if False:
+    import requests
+    from torchvision import transforms
+    model = torch.hub.load("pytorch/vision:v0.6.0", "resnet18", pretrained=True).eval()
+    response = requests.get("https://git.io/JJkYN")
+    labels = response.text.split("\n")
 
 
 def predict(inp, *args, **kwargs):