leolaish commited on
Commit
a328976
·
verified ·
1 Parent(s): 3eba1c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -552,11 +552,11 @@ def generate(message, temperature):
552
  If an error occurred, the boolean flag will be True and the content will be the error message.
553
  """
554
  stream = client.chat.completions.create(
555
- model="tgi",
556
  messages=message,
557
  stream=False,
558
  max_tokens=1024,
559
- stop=["```output\n"],
560
  temperature=temperature,
561
  #timeout=30,
562
  )
 
552
  If an error occurred, the boolean flag will be True and the content will be the error message.
553
  """
554
  stream = client.chat.completions.create(
555
+ model="H4/zephyr-7b-beta",
556
  messages=message,
557
  stream=False,
558
  max_tokens=1024,
559
+ stop=["output\n"],
560
  temperature=temperature,
561
  #timeout=30,
562
  )