shivi commited on
Commit
6230ced
·
verified ·
1 Parent(s): c06731e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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='c4ai-aya-23-35b', connectors=[], temperature=0.3)
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