kaitwithkwk commited on
Commit
0674565
·
verified ·
1 Parent(s): dcfbd74

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -4
app.py CHANGED
@@ -11,13 +11,19 @@ def respond(message, history):
11
  messages.extend(history)
12
 
13
  messages.append({"role": "user", "content": message})
 
 
14
 
15
- response = client.chat_completion(
16
  messages,
17
- max_tokens=100
 
18
  )
19
-
20
- return response['choices'][0]['message']['content'].strip()
 
 
 
21
 
22
  chatbot = gr.ChatInterface(respond, type="messages")
23
 
 
11
  messages.extend(history)
12
 
13
  messages.append({"role": "user", "content": message})
14
+
15
+ response = ""
16
 
17
+ response_stream = client.chat_completion(
18
  messages,
19
+ max_tokens=100,
20
+ stream=True
21
  )
22
+
23
+ for token in response_stream:
24
+ token = message['choices'][0]['message']['content'].strip()
25
+ response += token
26
+ yield response
27
 
28
  chatbot = gr.ChatInterface(respond, type="messages")
29