Update app.py

app.py

@@ -8,7 +8,7 @@ import copy
 
 # Default configuration
 default_config = {
-    "main_model": "llama-3.1-70b-versatile",
+    "main_model": "llama-3.2-3b-preview",
     "cycles": 3,
     "layer_agent_config": {}
 }
@@ -20,7 +20,7 @@ layer_agent_config_def = {
     },
     "layer_agent_2": {
         "system_prompt": "Respond with a thought and then your response to the question. {helper_response}",
-        "model_name": "gemma-7b-it",
+        "model_name": "llama-3.2-3b-preview",
         "temperature": 0.7
     },
     "layer_agent_3": {
@@ -33,7 +33,7 @@ layer_agent_config_def = {
 # Recommended Configuration
 
 rec_config = {
-    "main_model": "llama-3.2-90b-text-preview",
+    "main_model": "llama-3.2-3b-preview",
     "cycles": 2,
     "layer_agent_config": {}
 }
@@ -134,14 +134,12 @@ st.set_page_config(
     layout="wide"
 )
 valid_model_names = [
-    'llama-3.2-90b-text-preview',
     'llama3-70b-8192',
     'llama3-8b-8192',
-    'llama-3.1-70b-versatile',
+    'llama-3.3-70b-versatile',
     'llama-3.1-8b-instant',
-    'llama3-groq-70b-8192-tool-use-preview',
-    'llama3-groq-8b-8192-tool-use-preview',
-    'gemma-7b-it',
+    'llama-3.2-3b-preview',
+    'llama-3.2-1b-preview',
     'gemma2-9b-it',
     'mixtral-8x7b-32768'
 ]