wop commited on
Commit
71f38ca
·
verified ·
1 Parent(s): 6a8a2c3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -6
app.py CHANGED
@@ -7,7 +7,7 @@ import datetime
7
  import json
8
 
9
  _ = load_dotenv(find_dotenv())
10
- st.set_page_config(page_icon="💬", layout="wide", page_title="Groq Chat Bot...")
11
 
12
 
13
  def icon(emoji: str):
@@ -18,15 +18,15 @@ def icon(emoji: str):
18
  )
19
 
20
 
21
- icon("📣")
22
 
23
- st.subheader("Groq Chat Streamlit App", divider="rainbow", anchor=False)
24
 
25
  client = Groq(
26
  api_key=os.environ['GROQ_API_KEY'],
27
  )
28
 
29
- def get_tool_owner_info():
30
  owner_info = {
31
  "date_time": datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
32
  }
@@ -56,7 +56,7 @@ def run_conversation(user_prompt, messages, model_option, max_tokens):
56
 
57
  if tool_calls:
58
  available_functions = {
59
- "time_date": get_tool_owner_info
60
  }
61
 
62
  messages.append(response_message)
@@ -99,8 +99,10 @@ models = {
99
  "tokens": 32768,
100
  "developer": "Mistral",
101
  },
102
- "llama2-70b-4096": {"name": "LLaMA2-70b-chat", "tokens": 4096, "developer": "Meta"},
103
  "gemma-7b-it": {"name": "Gemma-7b-it", "tokens": 8192, "developer": "Google"},
 
 
 
104
  }
105
 
106
  # Layout for model selection and max_tokens slider
 
7
  import json
8
 
9
  _ = load_dotenv(find_dotenv())
10
+ st.set_page_config(page_icon="💬", layout="wide", page_title="...")
11
 
12
 
13
  def icon(emoji: str):
 
18
  )
19
 
20
 
21
+ icon("")
22
 
23
+ st.subheader("Chatbot", divider="rainbow", anchor=False)
24
 
25
  client = Groq(
26
  api_key=os.environ['GROQ_API_KEY'],
27
  )
28
 
29
+ def get_time_date_tool():
30
  owner_info = {
31
  "date_time": datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
32
  }
 
56
 
57
  if tool_calls:
58
  available_functions = {
59
+ "time_date": get_time_date_tool
60
  }
61
 
62
  messages.append(response_message)
 
99
  "tokens": 32768,
100
  "developer": "Mistral",
101
  },
 
102
  "gemma-7b-it": {"name": "Gemma-7b-it", "tokens": 8192, "developer": "Google"},
103
+ "llama2-70b-4096": {"name": "LLaMA2-70b-chat", "tokens": 4096, "developer": "Meta"},
104
+ "llama3-70b-8192": {"name": "LLaMA3-70b-8192", "tokens": 8192, "developer": "Meta"},
105
+ "llama3-8b-8192": {"name": "LLaMA3-8b-8192", "tokens": 8192, "developer": "Meta"},
106
  }
107
 
108
  # Layout for model selection and max_tokens slider