Update app.py

app.py

@@ -9,7 +9,7 @@ import copy
 
 # Default configuration
 default_config = {
-    "main_model": "llama3-70b-8192",
+    "main_model": "llama-3.1-70b-versatile",
     "cycles": 3,
     "layer_agent_config": {}
 }
@@ -17,16 +17,16 @@ default_config = {
 layer_agent_config_def = {
     "layer_agent_1": {
         "system_prompt": "Think through your response step by step. {helper_response}",
-        "model_name": "llama3-8b-8192"
+        "model_name": "llama-3.1-8b-instant"
     },
     "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": "gemma2-9b-it",
         "temperature": 0.7
     },
     "layer_agent_3": {
         "system_prompt": "You are an expert at logic and reasoning. Always take a logical approach to the answer. {helper_response}",
-        "model_name": "llama3-8b-8192"
+        "model_name": "llama3-70b-8192"
     },
 
 }
@@ -34,7 +34,7 @@ layer_agent_config_def = {
 # Recommended Configuration
 
 rec_config = {
-    "main_model": "llama3-70b-8192",
+    "main_model": "llama-3.1-70b-versatile",
     "cycles": 2,
     "layer_agent_config": {}
 }
@@ -42,17 +42,17 @@ rec_config = {
 layer_agent_config_rec = {
     "layer_agent_1": {
         "system_prompt": "Think through your response step by step. {helper_response}",
-        "model_name": "llama3-8b-8192",
+        "model_name": "llama-3.1-8b-instant",
         "temperature": 0.1
     },
     "layer_agent_2": {
         "system_prompt": "Respond with a thought and then your response to the question. {helper_response}",
-        "model_name": "llama3-8b-8192",
+        "model_name": "gemma2-9b-it",
         "temperature": 0.2
     },
     "layer_agent_3": {
         "system_prompt": "You are an expert at logic and reasoning. Always take a logical approach to the answer. {helper_response}",
-        "model_name": "llama3-8b-8192",
+        "model_name": "llama3-70b-8192",
         "temperature": 0.4
     },
     "layer_agent_4": {
@@ -226,6 +226,7 @@ with st.sidebar:
     - MOA: [Together AI](https://www.together.ai/blog/together-moa)
     - LLMs: [Groq](https://groq.com/)
     - Paper: [arXiv:2406.04692](https://arxiv.org/abs/2406.04692)
+    - Code: https://github.com/skapadia3214/groq-moa
     """)
 
 # Main app layout