silterra commited on
Commit
6862690
·
1 Parent(s): 43db59c

Simplify Dockerfile. Just testing streamlit now

Browse files
Files changed (2) hide show
  1. Dockerfile +20 -5
  2. app.py +0 -2
Dockerfile CHANGED
@@ -1,13 +1,28 @@
1
- # Use the same container for the HF-space that we compiled for general use
2
- FROM silterra/diffdock-pocket-dev
 
 
 
 
 
 
 
 
 
 
3
  USER $APPUSER
 
 
 
 
 
4
 
5
- COPY --chown=$APPUSER:appgroup . /home/$APPUSER/app
6
 
7
- WORKDIR /home/$APPUSER/app
8
 
9
  # Expose port for streamlit
10
  EXPOSE 8501
11
 
12
  # Run streamlit app under conda environment
13
- CMD ["sh", "-c", "micromamba run -n ${ENV_NAME} streamlit run app.py"]
 
1
+ FROM nvidia/cuda:11.7.1-devel-ubuntu22.04
2
+
3
+ RUN apt-get update -y
4
+ RUN apt-get install -y sudo wget curl nano git \
5
+ python3 python3-pip && rm -rf /var/lib/apt/lists/*
6
+
7
+ # Create a group and user
8
+ ENV APPUSER="appuser"
9
+ ENV HOME=/home/$APPUSER
10
+ RUN groupadd -g 999 appgroup && \
11
+ useradd -r -u 999 -g appgroup $APPUSER
12
+
13
  USER $APPUSER
14
+ WORKDIR $HOME
15
+
16
+ RUN pip install --user streamlit==1.28
17
+
18
+ ENV PATH="$HOME/.local/bin:$PATH"
19
 
20
+ COPY --chown=$APPUSER:appgroup . $HOME/app
21
 
22
+ WORKDIR $HOME/app
23
 
24
  # Expose port for streamlit
25
  EXPOSE 8501
26
 
27
  # Run streamlit app under conda environment
28
+ CMD ["sh", "-c", "streamlit run app.py"]
app.py CHANGED
@@ -1,5 +1,3 @@
1
- import torch
2
-
3
  import streamlit as st
4
 
5
  # Create the Streamlit app
 
 
 
1
  import streamlit as st
2
 
3
  # Create the Streamlit app