vincentmin commited on
Commit
5385efe
1 Parent(s): ddaf3fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -8
app.py CHANGED
@@ -65,7 +65,7 @@ def chat_accordion():
65
  return model, temperature, top_p
66
 
67
 
68
- def format_chat_prompt(message: str, chat_history, bot_name: str, instructions: str) -> str:
69
  print(bot_name)
70
  instructions = INSTRUCTIONS_MAPPING[bot_name].strip(" ").strip("\n")
71
  prompt = instructions
@@ -119,9 +119,8 @@ def chat():
119
  interactive=True,
120
  label="Character",
121
  )
122
- instructions=''
123
 
124
- def run_chat(message: str, chat_history, bot_name: str, instructions: str, model: str, temperature: float, top_p: float):
125
  if not message or (message == RETRY_COMMAND and len(chat_history) == 0):
126
  yield chat_history
127
  return
@@ -131,7 +130,7 @@ def chat():
131
  user_message, _ = prev_turn
132
  message = user_message
133
 
134
- prompt = format_chat_prompt(message, chat_history, bot_name, instructions)
135
  model_output = run_model(
136
  prompt,
137
  model=model,
@@ -148,15 +147,15 @@ def chat():
148
  chat_history.pop(-1)
149
  return {chatbot: gr.update(value=chat_history)}
150
 
151
- def run_retry(message: str, chat_history, bot_name, instructions: str, model: str, temperature: float, top_p: float):
152
- yield from run_chat(RETRY_COMMAND, chat_history, bot_name, instructions, model, temperature, top_p)
153
 
154
  def clear_chat():
155
  return []
156
 
157
  inputs.submit(
158
  run_chat,
159
- [inputs, chatbot, bot_name, instructions, model, temperature, top_p],
160
  outputs=[chatbot],
161
  show_progress=False,
162
  )
@@ -164,7 +163,7 @@ def chat():
164
  delete_turn_button.click(delete_last_turn, inputs=[chatbot], outputs=[chatbot])
165
  retry_button.click(
166
  run_retry,
167
- [inputs, chatbot, bot_name, instructions, model, temperature, top_p],
168
  outputs=[chatbot],
169
  show_progress=False,
170
  )
 
65
  return model, temperature, top_p
66
 
67
 
68
+ def format_chat_prompt(message: str, chat_history, bot_name: str) -> str:
69
  print(bot_name)
70
  instructions = INSTRUCTIONS_MAPPING[bot_name].strip(" ").strip("\n")
71
  prompt = instructions
 
119
  interactive=True,
120
  label="Character",
121
  )
 
122
 
123
+ def run_chat(message: str, chat_history, bot_name: str, model: str, temperature: float, top_p: float):
124
  if not message or (message == RETRY_COMMAND and len(chat_history) == 0):
125
  yield chat_history
126
  return
 
130
  user_message, _ = prev_turn
131
  message = user_message
132
 
133
+ prompt = format_chat_prompt(message, chat_history, bot_name)
134
  model_output = run_model(
135
  prompt,
136
  model=model,
 
147
  chat_history.pop(-1)
148
  return {chatbot: gr.update(value=chat_history)}
149
 
150
+ def run_retry(message: str, chat_history, bot_name, model: str, temperature: float, top_p: float):
151
+ yield from run_chat(RETRY_COMMAND, chat_history, bot_name, model, temperature, top_p)
152
 
153
  def clear_chat():
154
  return []
155
 
156
  inputs.submit(
157
  run_chat,
158
+ [inputs, chatbot, bot_name, model, temperature, top_p],
159
  outputs=[chatbot],
160
  show_progress=False,
161
  )
 
163
  delete_turn_button.click(delete_last_turn, inputs=[chatbot], outputs=[chatbot])
164
  retry_button.click(
165
  run_retry,
166
+ [inputs, chatbot, bot_name, model, temperature, top_p],
167
  outputs=[chatbot],
168
  show_progress=False,
169
  )