rwitz commited on
Commit
64776a6
·
verified ·
1 Parent(s): e03abb4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -98,7 +98,7 @@ async def get_bot_response(adapter_id, prompt, state, bot_index):
98
  "parameters": {
99
  "adapter_id": adapter_id,
100
  "adapter_source": "hub",
101
- "temperature": 0.01,
102
  "max_new_tokens": 100
103
  }
104
  }
@@ -215,6 +215,8 @@ async def direct_chat(model, user_input, chatbot):
215
  chatbot.append((user_input, response))
216
  return "", chatbot
217
  # ...
 
 
218
  refresh_leaderboard()
219
  # Gradio interface setup
220
  # Gradio interface setup
@@ -247,6 +249,10 @@ with gr.Blocks() as demo:
247
  direct_textbox = gr.Textbox(placeholder="Enter your message")
248
  direct_submit_btn = gr.Button(value="Send")
249
  direct_reset_btn = gr.Button(value="Reset Chat") # Add this line
 
 
 
 
250
  with gr.Tab("Leaderboard"):
251
  try:
252
  leaderboard = gr.Dataframe(refresh_leaderboard())
 
98
  "parameters": {
99
  "adapter_id": adapter_id,
100
  "adapter_source": "hub",
101
+ "temperature": 0.7,
102
  "max_new_tokens": 100
103
  }
104
  }
 
215
  chatbot.append((user_input, response))
216
  return "", chatbot
217
  # ...
218
+ def reset_direct_chat():
219
+ return "", [], gr.Dropdown.update(value=model_dropdown.value)
220
  refresh_leaderboard()
221
  # Gradio interface setup
222
  # Gradio interface setup
 
249
  direct_textbox = gr.Textbox(placeholder="Enter your message")
250
  direct_submit_btn = gr.Button(value="Send")
251
  direct_reset_btn = gr.Button(value="Reset Chat") # Add this line
252
+
253
+ direct_textbox.submit(direct_chat, inputs=[model_dropdown, direct_textbox, direct_chatbot], outputs=[direct_textbox, direct_chatbot])
254
+ direct_submit_btn.click(direct_chat, inputs=[model_dropdown, direct_textbox, direct_chatbot], outputs=[direct_textbox, direct_chatbot])
255
+ direct_reset_btn.click(reset_direct_chat, None, [direct_textbox, direct_chatbot, model_dropdown])
256
  with gr.Tab("Leaderboard"):
257
  try:
258
  leaderboard = gr.Dataframe(refresh_leaderboard())