Update app.py
Browse files
app.py
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
import streamlit as st
|
2 |
-
import
|
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=
|
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
|