Update app.py

app.py

@@ -3,9 +3,6 @@ import gradio as gr
 
 from openai import OpenAI
 
-from optillm.moa import mixture_of_agents 
-from optillm.mcts import chat_with_mcts
-from optillm.bon import best_of_n_sampling
 from optillm.cot_reflection import cot_reflection
 
 API_KEY = os.environ.get("OPENROUTER_API_KEY")
@@ -34,7 +31,7 @@ def respond(
 
     # response = ""
 
-    final_response = mixture_of_agents(system_message, message, client, model)
+    final_response = cot_reflection(system_message, message, client, model)
     return final_response
 
     # for message in client.chat_completion(
@@ -61,7 +58,7 @@ demo = gr.ChatInterface(
             value="nousresearch/hermes-3-llama-3.1-405b:free", label="Model", info="Choose the base model"
         ),
         gr.Dropdown(
-            ["bon", "mcts", "moa", "cot_reflection"], value="moa", label="Approach", info="Choose the approach"
+            ["leap", "plansearch", "rstar", "cot_reflection"], value="cot_reflection", label="Approach", info="Choose the approach"
         ),
         gr.Textbox(value="", label="System message"),
         gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),