Update app.py

app.py

@@ -68,7 +68,7 @@ def predict(
     elif prompt_manipulation == "Random RAG (indirect steering with randomized context)":
         with open(f"data/ids_{direct_steering_option}.json", "r") as file:
             ids = json.load(file)
-        random_ids = random.sample(ids, n_results)
+        random_ids = random.sample(ids, num_contexts)
         
         impersonation_template = ""
         answer_option_template = f"{test_format[ideology_test]}"
@@ -217,13 +217,13 @@ def main():
                 togetherai_api_key = gr.Textbox(label="Together.ai API Key", placeholder="Enter your Together.ai API key here", show_label=True, type="password")
 
             with gr.Row():
-                temp_input = gr.Slider(minimum=0, maximum=1, step=0.01, label="Temperature", value=0.7)
+                temp_input = gr.Slider(minimum=0, maximum=2, step=0.01, label="Temperature", value=0.7)
                 
             with gr.Row():
                 top_p_input = gr.Slider(minimum=0, maximum=1, step=0.01, label="Top P", value=1)
                 
             with gr.Row():
-                num_contexts = gr.Slider(minimum=0, maximum=1, step=0.01, label="Top k retrieved contexts", value=3)
+                num_contexts = gr.Slider(minimum=0, maximum=5, step=1, label="Top k retrieved contexts", value=3)
         
         # Link settings to the predict function
         submit_btn.click(