Spaces:
Running
Running
Commit
Browse files
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 |
-
|
| 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,
|
| 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":
|
| 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 |
-
|
| 97 |
-
final_response = ""
|
| 98 |
-
for r in response_gen:
|
| 99 |
-
|
| 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,
|