disLodge commited on
Commit
94b1c33
·
verified ·
1 Parent(s): a6161ba
Files changed (1) hide show
  1. app.py +9 -9
app.py CHANGED
@@ -18,7 +18,7 @@ def respond(
18
  messages = [{"role": "system", "content": system_message}] + history
19
  messages.append({"role":"user","content":message})
20
 
21
- reponse = ""
22
 
23
  # for val in history:
24
  # if val[0]:
@@ -27,7 +27,7 @@ def respond(
27
  # messages.append({"role": "assistant", "content": val[1]})
28
 
29
  for part in client.chat_completion(
30
- messages, max_tokens=max_tokens, streams=True, temperature=temperature,
31
  top_p=top_p
32
  ):
33
  token = part.choices[0].delta.content
@@ -35,7 +35,7 @@ def respond(
35
  response += token
36
 
37
  history.append({"role":"user", "content": message})
38
- history.append({"role":"assistant", "content": respond})
39
 
40
  return history,""
41
 
@@ -93,12 +93,12 @@ with gr.Blocks() as demo:
93
  dummy_btn = gr.Button("Dummy Action")
94
 
95
  def handle_submit(message, history, system, max_tokens, temperature, top_p):
96
- response_gen = response(message, history, system, max_tokens, temperature, top_p)
97
- final_response = ""
98
- for r in response_gen:
99
- final_response = r
100
- updated_history = history + [(message, final_response)]
101
- return updated_history, updated_history, ""
102
 
103
  send_btn.click(
104
  handle_submit,
 
18
  messages = [{"role": "system", "content": system_message}] + history
19
  messages.append({"role":"user","content":message})
20
 
21
+ response = ""
22
 
23
  # for val in history:
24
  # if val[0]:
 
27
  # messages.append({"role": "assistant", "content": val[1]})
28
 
29
  for part in client.chat_completion(
30
+ messages, max_tokens=max_tokens, stream=True, temperature=temperature,
31
  top_p=top_p
32
  ):
33
  token = part.choices[0].delta.content
 
35
  response += token
36
 
37
  history.append({"role":"user", "content": message})
38
+ history.append({"role":"assistant", "content": response})
39
 
40
  return history,""
41
 
 
93
  dummy_btn = gr.Button("Dummy Action")
94
 
95
  def handle_submit(message, history, system, max_tokens, temperature, top_p):
96
+ return respond(message, history, system, max_tokens, temperature, top_p)
97
+ # final_response = ""
98
+ # for r in response_gen:
99
+ # final_response = r
100
+ # updated_history = history + [(message, final_response)]
101
+ # return updated_history, updated_history, ""
102
 
103
  send_btn.click(
104
  handle_submit,