Update app.py

app.py

@@ -45,14 +45,14 @@ class cbfs:
             chosen_llm = ChatOpenAI(model_name="gpt-4-1106-preview", temperature=0)
         # elif model_name == "GPT-3.5":
         #     chosen_llm = ChatOpenAI(model_name="gpt-3.5-turbo-0125", temperature=0)
-        elif model_name == "Llama-3 8B":
-            chosen_llm = ChatOpenRouter(model_name="meta-llama/llama-3-8b-instruct", temperature=0)
-        elif model_name == "Gemini-1.5 Pro":
-            chosen_llm = ChatOpenRouter(model_name="google/gemini-pro-1.5", temperature=0)
-        elif model_name == "Claude 3 Sonnet":
-            chosen_llm = ChatOpenRouter(model_name='anthropic/claude-3-sonnet', temperature=0)
-        elif model_name == "Claude 3.5 Sonnet":
-            chosen_llm = ChatOpenRouter(model_name='anthropic/claude-3.5-sonnet', temperature=0)
+        # elif model_name == "Llama-3 8B":
+        #     chosen_llm = ChatOpenRouter(model_name="meta-llama/llama-3-8b-instruct", temperature=0)
+        # elif model_name == "Gemini-1.5 Pro":
+        #     chosen_llm = ChatOpenRouter(model_name="google/gemini-pro-1.5", temperature=0)
+        # elif model_name == "Claude 3 Sonnet":
+        #     chosen_llm = ChatOpenRouter(model_name='anthropic/claude-3-sonnet', temperature=0)
+        # elif model_name == "Claude 3.5 Sonnet":
+        #     chosen_llm = ChatOpenRouter(model_name='anthropic/claude-3.5-sonnet', temperature=0)
         else:
             # Default model
             # chosen_llm = ChatOpenRouter(model_name="meta-llama/llama-3-70b-instruct", temperature=0)
@@ -122,10 +122,10 @@ with gr.Blocks() as demo:
 
     with gr.Row():
         db_choice = gr.Dropdown(["Covenants", "Bylaws"], label="Select Document", scale=1)
-        model_choice = gr.Dropdown(["GPT-3.5", "GPT-4", "Llama-3 70B", "Llama-3 8B", "Gemini-1.5 Pro", "Claude 3 Sonnet", "Claude 3.5 Sonnet"], 
-                                   label="Select Model", scale=1, value = "Llama-3 70B")
-        # model_choice = gr.Dropdown(["GPT-3.5", "GPT-4"], 
-        #                            label="Select Model", scale=1, value = "GPT-3.5")
+        # model_choice = gr.Dropdown(["GPT-3.5", "GPT-4", "Llama-3 70B", "Llama-3 8B", "Gemini-1.5 Pro", "Claude 3 Sonnet", "Claude 3.5 Sonnet"], 
+        #                            label="Select Model", scale=1, value = "Llama-3 70B")
+        model_choice = gr.Dropdown(["GPT-3.5", "GPT-4"], 
+                                   label="Select Model", scale=1, value = "GPT-3.5")
         button_clearhistory = gr.Button("Clear History", scale=1)
 
     with gr.Row():