Update PolyAgent/gradio_interface.py

PolyAgent/gradio_interface.py

@@ -1218,7 +1218,7 @@ def llm_only_answer(state: Dict[str, Any], model_name: str, prompt: str) -> str:
         )
 
     HF_MODEL_MAP = {
-        "mixtral-8x7b-instruct": "mistralai/Mixtral-8x7B-Instruct-v0.1",
+        "mixtral-8x22b-instruct": "mistralai/Mixtral-8x22B-Instruct-v0.1",
         "llama-3.1-8b-instruct": "meta-llama/Llama-3.1-8B-Instruct",
     }
 
@@ -1385,8 +1385,8 @@ def build_ui() -> gr.Blocks:
 
                 llm_model = gr.Dropdown(
                     label="Model",
-                    choices=["mixtral-8x7b-instruct", "llama-3.1-8b-instruct"],
-                    value="mixtral-8x7b-instruct",
+                    choices=["mixtral-8x22b-instruct", "llama-3.1-8b-instruct"],
+                    value="mixtral-8x22b-instruct",
                 )
                 llm_prompt = gr.Textbox(label="Prompt", lines=10, placeholder="Enter your polymer question/prompt.")
                 llm_btn = gr.Button("Run LLM", variant="primary")