Skip to content

Commit 200a8bd

Browse files
update controller
1 parent 9ca7419 commit 200a8bd

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

src/agentlab/analyze/agent_controller.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,10 @@
2020
from transformers import AutoTokenizer
2121

2222
# used to display prompt. simple chat template from apache 2.0 model
23-
tokenizer = AutoTokenizer.from_pretrained("HuggingFaceH4/zephyr-7b-beta")
23+
# tokenizer = AutoTokenizer.from_pretrained("HuggingFaceH4/zephyr-7b-beta")
24+
tokenizer = AutoTokenizer.from_pretrained(
25+
"/Users/patrice.bechard/.cache/huggingface/hub/models--HuggingFaceH4--zephyr-7b-beta/snapshots/892b3d7a7b1cf10c7a701c60881cd93df615734c"
26+
)
2427

2528
logger = logging.getLogger(__name__)
2629
logger.setLevel(logging.INFO)
@@ -803,7 +806,7 @@ def set_next_step_section():
803806

804807
def set_controller():
805808
with st.container(border=True):
806-
st.markdown("##### 🎮 Control")
809+
st.markdown("##### 🕹️ Control")
807810
set_agent_state_box()
808811
set_prompt_modifier()
809812
col_prev, col_redo, col_next = st.columns([1, 1, 1])
@@ -946,12 +949,12 @@ def run_streamlit():
946949
# config page
947950
st.set_page_config(
948951
page_title="AgentLab Controller",
949-
page_icon="🎮",
952+
page_icon="🕹️",
950953
layout="wide",
951954
initial_sidebar_state="collapsed",
952955
)
953956
st.markdown(
954-
'<h1 style="text-align: center;">🎮 AgentLab Controller 🎮</h1>', unsafe_allow_html=True
957+
'<h1 style="text-align: center;">🕹️ AgentLab Controller 🕹️</h1>', unsafe_allow_html=True
955958
)
956959

957960
setup_sidebar()

0 commit comments

Comments
 (0)