Update app.py

app.py

@@ -43,20 +43,20 @@ class cbfs:
         # Select model dynamically
         if model_name == "GPT-4":
             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)
-        else:
+        # 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)
+        # else:
             # Default model
-            chosen_llm = ChatOpenRouter(model_name="meta-llama/llama-3-70b-instruct", temperature=0)
-            # chosen_llm = ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0)
+            # chosen_llm = ChatOpenRouter(model_name="meta-llama/llama-3-70b-instruct", temperature=0)
+            chosen_llm = ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0)
 
         self.qa = ConversationalRetrievalChain.from_llm(
             llm=chosen_llm,
@@ -122,8 +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", "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():