asdf

Dockerfile

@@ -15,9 +15,9 @@ RUN pip install --no-cache-dir --upgrade pip && \
     chown -R user:user $HOME
 
 EXPOSE 7860
-CMD huggingface-cli login --token $HF_TOKEN --add-to-git-credential && \
-    streamlit run app.py \
-        --server.headless true \
-        --server.enableCORS false \
-        --server.enableXsrfProtection false \
-        --server.fileWatcherType none \
+# CMD huggingface-cli login --token $HF_TOKEN --add-to-git-credential --write-permission
+CMD streamlit run app.py \
+    --server.headless true \
+    --server.enableCORS false \
+    --server.enableXsrfProtection false \
+    --server.fileWatcherType none \

agent.py

@@ -107,6 +107,16 @@ 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