Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-import secrets
+import os
 
 #from transformers import BertModel, BertTokenizer
 from transformers import HfAgent, load_tool
@@ -74,7 +74,7 @@ class CustomHfAgent(Agent):
     """
 
     def __init__(
-        self, url_endpoint, token=secrets.HF_token, chat_prompt_template=None, run_prompt_template=None, additional_tools=None
+        self, url_endpoint, token=os.environ['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