Artin2009 commited on
Commit
9d8cbe0
1 Parent(s): 95537a7

Update chain_app.py

Browse files
Files changed (1) hide show
  1. chain_app.py +5 -3
chain_app.py CHANGED
@@ -1602,9 +1602,11 @@ async def main(message: cl.Message):
1602
  elif chat_profile == 'mistral-nemo-12B':
1603
  client = Client("0x7o/Mistral-Nemo-Instruct", hf_token=hf_token)
1604
  result = client.predict(
1605
- query=message.content,
1606
- system="You are a helpful AI chatbot made by two iranian boys named Artin Daneshvar and Sadra Noadoust",
1607
- api_name="/chat"
 
 
1608
  )
1609
  await cl.Message(
1610
  content=result[1][0][1]
 
1602
  elif chat_profile == 'mistral-nemo-12B':
1603
  client = Client("0x7o/Mistral-Nemo-Instruct", hf_token=hf_token)
1604
  result = client.predict(
1605
+ message=message.content,
1606
+ max_new_tokens=512,
1607
+ temperature=0.7,
1608
+ top_p=0.95,
1609
+ api_name="/chat"
1610
  )
1611
  await cl.Message(
1612
  content=result[1][0][1]