update

pages/1_Injection.py

@@ -27,7 +27,7 @@ def initialize_state():
     keys = ["model_submitted", "api_key", "endpoint_url", "deployment_name", "temperature", "max_tokens",
             "data_processed", "group_name", "occupation", "privilege_label", "protect_label", "num_run",
             "uploaded_file", "additional_charateristics"]
-    defaults = [False, "", "https://safeguard-monitor.openai.azure.com/", "gpt35-1106", 0.5, 150, False, "Gender",
+    defaults = [False, "", "https://safeguard-monitor.openai.azure.com/", "gpt35-1106", 0.0, 150, False, "Gender",
                 "Programmer", "Male", "Female", 1, None, None]
     for key, default in zip(keys, defaults):
         if key not in st.session_state:
@@ -48,7 +48,7 @@ else:
     st.session_state.endpoint_url = st.sidebar.text_input("Endpoint URL", value=st.session_state.endpoint_url)
     st.session_state.deployment_name = st.sidebar.text_input("Model Name", value=st.session_state.deployment_name)
     api_version = '2024-02-15-preview' if model_type == 'GPTAgent' else ''
-    st.session_state.temperature = st.sidebar.slider("Temperature", 0.0, 1.0, st.session_state.temperature, 0.00)
+    st.session_state.temperature = st.sidebar.slider("Temperature", 0.0, 1.0, st.session_state.temperature, 0.01)
     st.session_state.max_tokens = st.sidebar.number_input("Max Tokens", 1, 1000, st.session_state.max_tokens)
 
     if st.sidebar.button("Reset Model Info"):