update

Dockerfile

@@ -1,27 +1,29 @@
 FROM ubuntu:22.04
 
-# Set Environment Variable
-USER root
-ENV HOME="/root"
+RUN useradd -m -s /bin/bash myuser
+ENV HOME=/home/myuser
+
 ENV JAVA_TOOL_OPTIONS="-Dhttps.protocols=TLSv1.2"
-ENV PDF2JSON_HOME="/app/src/s2orc-doc2json"
+ENV PDF2JSON_HOME=$HOME/app/src/s2orc-doc2json
 
 # install system-wide deps for python and node
 RUN apt-get -yqq update && \
     apt-get -yqq install software-properties-common curl wget zip screen git gcc build-essential openjdk-8-jdk 
 
+USER myuser
 # Install Miniconda
+WORKDIR $HOME
 RUN curl -LO https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh && \
-    bash Miniconda3-latest-Linux-x86_64.sh -p /miniconda -b && \
+    bash Miniconda3-latest-Linux-x86_64.sh -p $HOME/miniconda -b && \
     rm Miniconda3-latest-Linux-x86_64.sh
-ENV PATH=/miniconda/bin:${PATH}
+ENV PATH=$HOME/miniconda/bin:${PATH}
 
 # Create a Python 3.10 environment
 RUN conda create -n my_env python=3.10
 
 SHELL ["conda", "run", "-n", "my_env", "/bin/bash", "-c"]
 
-WORKDIR /app/src
+WORKDIR $HOME/app/src
 COPY ./requirements.txt .
 RUN pip install -r requirements.txt
 
@@ -39,14 +41,14 @@ RUN ./gradlew clean install && \
     cp $PDF2JSON_HOME/doc2json/grobid2json/grobid/config.yaml $HOME/grobid-0.6.1/grobid-service/config/config.yaml && \
     cp $PDF2JSON_HOME/doc2json/grobid2json/grobid/grobid.properties $HOME/grobid-0.6.1/grobid-home/config/grobid.properties
 
-WORKDIR /app/models/
+WORKDIR $HOME/app/models/
 # Download necessary model checkpoint
-RUN python -c "from huggingface_hub import snapshot_download; model_folder = '/app/models/'; snapshot_download('nianlong/memsum-word-embedding', local_dir = model_folder + 'word_embedding'); snapshot_download('nianlong/memsum-arxiv-summarization', local_dir = model_folder + 'memsum_arxiv' )"
+RUN python -c "from huggingface_hub import snapshot_download; model_folder = './'; snapshot_download('nianlong/memsum-word-embedding', local_dir = model_folder + 'word_embedding'); snapshot_download('nianlong/memsum-arxiv-summarization', local_dir = model_folder + 'memsum_arxiv' )"
 
-WORKDIR /app/src
+WORKDIR $HOME/app/src
 COPY ./Dockerfile . 
 
-WORKDIR /app/src/services
+WORKDIR $HOME/app/src/services
 RUN git clone https://github.com/nianlonggu/MemSum
 
 COPY ./services/ .

services/app.py

@@ -45,8 +45,8 @@ def summarize_paper(memsum, paper_info):
     
 @st.cache_resource
 def load_models():
-    memsum = MemSum(  "/app/models/memsum_arxiv/model.pt", 
-                  "/app/models/word_embedding/vocabulary_200dim.pkl", 
+    memsum = MemSum(  "/home/myuser/app/models/memsum_arxiv/model.pt", 
+                  "/home/myuser/app/models/word_embedding/vocabulary_200dim.pkl", 
                   gpu = None,  max_doc_len = 500  )
     return memsum