ogegadavis254
commited on
Commit
•
c211211
1
Parent(s):
ed8bd19
Update app.py
Browse files
app.py
CHANGED
@@ -21,11 +21,6 @@ def reset_conversation():
|
|
21 |
st.session_state.messages = []
|
22 |
return None
|
23 |
|
24 |
-
# Define the available models
|
25 |
-
model_links = {
|
26 |
-
"Mistral-7B": "mistralai/Mistral-7B-Instruct-v0.2"
|
27 |
-
}
|
28 |
-
|
29 |
# Set the temperature value directly in the code
|
30 |
temperature = 0.5
|
31 |
|
@@ -62,9 +57,8 @@ if prompt := st.chat_input("Type your message here..."):
|
|
62 |
response = client.chat.completions.create(
|
63 |
model=model_link,
|
64 |
messages=[
|
65 |
-
system_message,
|
66 |
-
{"role": m["role"], "content": m["content"]}
|
67 |
-
for m in st.session_state.messages
|
68 |
],
|
69 |
temperature=temperature,
|
70 |
max_tokens=3000,
|
|
|
21 |
st.session_state.messages = []
|
22 |
return None
|
23 |
|
|
|
|
|
|
|
|
|
|
|
24 |
# Set the temperature value directly in the code
|
25 |
temperature = 0.5
|
26 |
|
|
|
57 |
response = client.chat.completions.create(
|
58 |
model=model_link,
|
59 |
messages=[
|
60 |
+
system_message, # System message
|
61 |
+
*[{"role": m["role"], "content": m["content"]} for m in st.session_state.messages] # User and assistant messages
|
|
|
62 |
],
|
63 |
temperature=temperature,
|
64 |
max_tokens=3000,
|