Chris4K commited on
Commit
45f17fe
·
verified ·
1 Parent(s): cbbdd46

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import streamlit as st
2
- import secrets
3
 
4
  #from transformers import BertModel, BertTokenizer
5
  from transformers import HfAgent, load_tool
@@ -74,7 +74,7 @@ class CustomHfAgent(Agent):
74
  """
75
 
76
  def __init__(
77
- self, url_endpoint, token=secrets.HF_token, chat_prompt_template=None, run_prompt_template=None, additional_tools=None
78
  ):
79
  # super()._init_(self, url_endpoint, token=None, chat_prompt_template=None, run_prompt_template=None, additional_tools=None)
80
  self.url_endpoint = url_endpoint
 
1
  import streamlit as st
2
+ import os
3
 
4
  #from transformers import BertModel, BertTokenizer
5
  from transformers import HfAgent, load_tool
 
74
  """
75
 
76
  def __init__(
77
+ self, url_endpoint, token=os.environ['HF_token'], chat_prompt_template=None, run_prompt_template=None, additional_tools=None
78
  ):
79
  # super()._init_(self, url_endpoint, token=None, chat_prompt_template=None, run_prompt_template=None, additional_tools=None)
80
  self.url_endpoint = url_endpoint