Jyothikamalesh commited on
Commit
5d4663f
·
verified ·
1 Parent(s): 3ba965f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -32,14 +32,15 @@ async def respond(
32
  messages.append({"role": "user", "content": message})
33
 
34
  response = ""
35
- async for message in client.chat.completions.create(
36
  model="NousResearch/Hermes-3-Llama-3.1-8B",
37
  max_tokens=max_tokens,
38
  stream=True,
39
  temperature=temperature,
40
  top_p=top_p,
41
  messages=messages,
42
- ):
 
43
  token = message.choices[0].text
44
  response += token
45
  return response
 
32
  messages.append({"role": "user", "content": message})
33
 
34
  response = ""
35
+ stream = client.chat.completions.create(
36
  model="NousResearch/Hermes-3-Llama-3.1-8B",
37
  max_tokens=max_tokens,
38
  stream=True,
39
  temperature=temperature,
40
  top_p=top_p,
41
  messages=messages,
42
+ )
43
+ for message in stream.iterate():
44
  token = message.choices[0].text
45
  response += token
46
  return response