umair894 commited on
Commit
4c2a4e1
1 Parent(s): 9b07741

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -18,7 +18,7 @@ def icon(emoji: str):
18
 
19
  # icon("⚡️")
20
 
21
- st.subheader("Groq Chat with LLaMA3 App", divider="rainbow", anchor=False)
22
 
23
  client = Groq(
24
  api_key=os.environ['GROQ_API_KEY'],
@@ -35,13 +35,13 @@ if "selected_model" not in st.session_state:
35
  models = {
36
  "llama3-70b-8192": {"name": "LLaMA3-70b", "tokens": 8192, "developer": "Meta"},
37
  "llama3-8b-8192": {"name": "LLaMA3-8b", "tokens": 8192, "developer": "Meta"},
38
- "llama2-70b-4096": {"name": "LLaMA2-70b-chat", "tokens": 4096, "developer": "Meta"},
39
- "gemma-7b-it": {"name": "Gemma-7b-it", "tokens": 8192, "developer": "Google"},
40
- "mixtral-8x7b-32768": {
41
- "name": "Mixtral-8x7b-Instruct-v0.1",
42
- "tokens": 32768,
43
- "developer": "Mistral",
44
- },
45
  }
46
 
47
  # Layout for model selection and max_tokens slider
 
18
 
19
  # icon("⚡️")
20
 
21
+ st.subheader("Groq Chat with LLaMA3", divider="rainbow", anchor=False)
22
 
23
  client = Groq(
24
  api_key=os.environ['GROQ_API_KEY'],
 
35
  models = {
36
  "llama3-70b-8192": {"name": "LLaMA3-70b", "tokens": 8192, "developer": "Meta"},
37
  "llama3-8b-8192": {"name": "LLaMA3-8b", "tokens": 8192, "developer": "Meta"},
38
+ # "llama2-70b-4096": {"name": "LLaMA2-70b-chat", "tokens": 4096, "developer": "Meta"},
39
+ # "gemma-7b-it": {"name": "Gemma-7b-it", "tokens": 8192, "developer": "Google"},
40
+ # "mixtral-8x7b-32768": {
41
+ # "name": "Mixtral-8x7b-Instruct-v0.1",
42
+ # "tokens": 32768,
43
+ # "developer": "Mistral",
44
+ # },
45
  }
46
 
47
  # Layout for model selection and max_tokens slider