Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Update app.py
Browse files
app.py
CHANGED
@@ -8,6 +8,7 @@ import secrets
|
|
8 |
|
9 |
|
10 |
cohere_api_key = os.getenv("COHERE_API_KEY")
|
|
|
11 |
co = cohere.Client(cohere_api_key, client_name="huggingface-aya-23")
|
12 |
|
13 |
|
@@ -29,7 +30,7 @@ def generate_response(user_message, cid, token, history=None):
|
|
29 |
|
30 |
history.append(user_message)
|
31 |
|
32 |
-
stream = co.chat_stream(message=user_message, conversation_id=cid, model=
|
33 |
#stream = co.generate(prompt=user_message, model='c4ai-aya-23')
|
34 |
output = ""
|
35 |
|
|
|
8 |
|
9 |
|
10 |
cohere_api_key = os.getenv("COHERE_API_KEY")
|
11 |
+
AYA_MODEL_NAME = os.getenv("AYA_MODEL_NAME")
|
12 |
co = cohere.Client(cohere_api_key, client_name="huggingface-aya-23")
|
13 |
|
14 |
|
|
|
30 |
|
31 |
history.append(user_message)
|
32 |
|
33 |
+
stream = co.chat_stream(message=user_message, conversation_id=cid, model=AYA_MODEL_NAME, connectors=[], temperature=0.3)
|
34 |
#stream = co.generate(prompt=user_message, model='c4ai-aya-23')
|
35 |
output = ""
|
36 |
|