Spaces:
Running
Running
Zekun Wu
commited on
Commit
·
ad82313
1
Parent(s):
5f064a8
update
Browse files- pages/1_Injection.py +4 -2
pages/1_Injection.py
CHANGED
@@ -56,11 +56,13 @@ else:
|
|
56 |
st.session_state.api_key = st.sidebar.text_input("API Key", type="password", value=st.session_state.api_key)
|
57 |
st.session_state.deployment_name = st.sidebar.text_input("Model Name", value=st.session_state.deployment_name)
|
58 |
|
|
|
|
|
|
|
59 |
if model_type == 'GPTAgent' or model_type == 'AzureAgent':
|
60 |
st.session_state.endpoint_url = st.sidebar.text_input("Endpoint URL", value=st.session_state.endpoint_url)
|
61 |
api_version = '2024-02-15-preview' if model_type == 'GPTAgent' else ''
|
62 |
-
|
63 |
-
st.session_state.max_tokens = st.sidebar.number_input("Max Tokens", 1, 1000, st.session_state.max_tokens)
|
64 |
|
65 |
if st.sidebar.button("Reset Model Info"):
|
66 |
initialize_state() # Reset all state to defaults
|
|
|
56 |
st.session_state.api_key = st.sidebar.text_input("API Key", type="password", value=st.session_state.api_key)
|
57 |
st.session_state.deployment_name = st.sidebar.text_input("Model Name", value=st.session_state.deployment_name)
|
58 |
|
59 |
+
st.session_state.temperature = st.sidebar.slider("Temperature", 0.0, 1.0, st.session_state.temperature, 0.00)
|
60 |
+
st.session_state.max_tokens = st.sidebar.number_input("Max Tokens", 1, 1000, st.session_state.max_tokens)
|
61 |
+
|
62 |
if model_type == 'GPTAgent' or model_type == 'AzureAgent':
|
63 |
st.session_state.endpoint_url = st.sidebar.text_input("Endpoint URL", value=st.session_state.endpoint_url)
|
64 |
api_version = '2024-02-15-preview' if model_type == 'GPTAgent' else ''
|
65 |
+
|
|
|
66 |
|
67 |
if st.sidebar.button("Reset Model Info"):
|
68 |
initialize_state() # Reset all state to defaults
|