Zekun Wu commited on
Commit
ad82313
1 Parent(s): 5f064a8
Files changed (1) hide show
  1. 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
- st.session_state.temperature = st.sidebar.slider("Temperature", 0.0, 1.0, st.session_state.temperature, 0.01)
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