Update app.py

app.py

@@ -1,4 +1,6 @@
 import streamlit as st
+import secrets
+
 #from transformers import BertModel, BertTokenizer
 from transformers import HfAgent, load_tool
 
@@ -72,7 +74,7 @@ class CustomHfAgent(Agent):
     """
 
     def __init__(
-        self, url_endpoint, token=None, chat_prompt_template=None, run_prompt_template=None, additional_tools=None
+        self, url_endpoint, token=secrets.HF_token, chat_prompt_template=None, run_prompt_template=None, additional_tools=None
     ):
        # super()._init_(self, url_endpoint, token=None, chat_prompt_template=None, run_prompt_template=None, additional_tools=None)
         self.url_endpoint = url_endpoint