asdf

Dockerfile

@@ -3,20 +3,20 @@ FROM python:3.12
 WORKDIR /app
 COPY ./requirements.txt /app/requirements.txt
 RUN pip install --no-cache-dir --upgrade -r /app/requirements.txt
+ARG HF_TOKEN
+RUN huggingface-cli login --token $HF_TOKEN --add-to-git-credential
 
 # Add user
 RUN useradd -m -u 1000 user
 USER user
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH
-# RUN chown -R user:user $HOME
-
 WORKDIR $HOME/app
 COPY --chown=user . $HOME
+RUN chown -R user:user $HOME
 
 EXPOSE 7860
-CMD huggingface-cli login --token $HF_TOKEN --add-to-git-credential && \
-    streamlit run app.py \
+CMD streamlit run app.py \
     --server.headless true \
     --server.enableCORS false \
     --server.enableXsrfProtection false \

agent.py

@@ -107,16 +107,6 @@ new_chat_template = """{{- bos_token }}
     {{- '<|start_header_id|>assistant<|end_header_id|>\n\n' }}
 {%- endif %}"""
 
-import os
-from huggingface_hub import login
-HF_KEY = os.getenv('HF_TOKEN')
-
-login(
-    token=HF_KEY,
-    add_to_git_credential=True,
-    write_permission=True
-)
-
 class Naomi:
     def __init__(self, **kwargs):
         # init dataclasses