Update app.py
Browse files
app.py
CHANGED
@@ -64,6 +64,15 @@ async def respond(
|
|
64 |
print(f"Error: {e}")
|
65 |
return "Error occurred. Please try again."
|
66 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
67 |
def launch_app():
|
68 |
try:
|
69 |
demo = gr.Blocks()
|
@@ -77,12 +86,7 @@ def launch_app():
|
|
77 |
top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-P")
|
78 |
response = gr.Text(label="Response")
|
79 |
|
80 |
-
|
81 |
-
new_history = history + [[message, ""]]
|
82 |
-
response = asyncio.run(respond(message, history, system_message, max_tokens, temperature, top_p))
|
83 |
-
new_history[-1][1] = response
|
84 |
-
return response, new_history
|
85 |
-
|
86 |
gr.Button("Generate Response").click(
|
87 |
generate_response,
|
88 |
inputs=[message, history, system_message, max_tokens, temperature, top_p],
|
|
|
64 |
print(f"Error: {e}")
|
65 |
return "Error occurred. Please try again."
|
66 |
|
67 |
+
|
68 |
+
# Make the Gradio app async
|
69 |
+
async def generate_response(message, history, system_message, max_tokens, temperature, top_p):
|
70 |
+
new_history = history + [[message, ""]]
|
71 |
+
response = await respond(message, history, system_message, max_tokens, temperature, top_p)
|
72 |
+
new_history[-1][1] = response
|
73 |
+
return response, new_history
|
74 |
+
|
75 |
+
|
76 |
def launch_app():
|
77 |
try:
|
78 |
demo = gr.Blocks()
|
|
|
86 |
top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-P")
|
87 |
response = gr.Text(label="Response")
|
88 |
|
89 |
+
# Use the async version of generate_response
|
|
|
|
|
|
|
|
|
|
|
90 |
gr.Button("Generate Response").click(
|
91 |
generate_response,
|
92 |
inputs=[message, history, system_message, max_tokens, temperature, top_p],
|