kevinwang676 commited on
Commit
75b94c9
1 Parent(s): 2355aad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -107,7 +107,7 @@ def predict(input, chatbot, max_length, top_p, temperature, history, past_key_va
107
  temperature=temperature):
108
  chatbot[-1] = (parse_text(input), parse_text(response))
109
 
110
- yield chatbot, history, past_key_values
111
 
112
 
113
  def reset_user_input():
@@ -226,22 +226,22 @@ if __name__ == "__main__":
226
  max_length = gr.Slider(0, 32768, value=8192, step=1.0, label="Maximum length", interactive=True)
227
  top_p = gr.Slider(0, 1, value=0.8, step=0.01, label="Top P", interactive=True)
228
  temperature = gr.Slider(0, 1, value=0.95, step=0.01, label="Temperature", interactive=True)
 
229
 
230
  history = gr.State([])
231
  past_key_values = gr.State(None)
232
 
233
  submitBtn.click(predict, [user_input, chatbot, max_length, top_p, temperature, history, past_key_values],
234
- [chatbot, history, past_key_values], show_progress=True)
235
  submitBtn.click(reset_user_input, [], [user_input])
236
 
237
- emptyBtn.click(reset_state, outputs=[chatbot, history, past_key_values], show_progress=True)
238
 
239
 
240
  with gr.Row():
241
  with gr.Column():
242
- text = gr.TextArea(label="Text", placeholder="Input Text Here",
243
- value="那是流萤吗,是明灭迷离,天真绮丽的憧憬。那是尘埃吧,是虚无纷飞,终将落地的谎言。")
244
- speaker = gr.Dropdown(choices=speakers, value=speakers[0], label='Speaker')
245
  with gr.Row():
246
  sdp_ratio = gr.Slider(minimum=0, maximum=1, value=0.2, step=0.1, label='语调变化')
247
  noise_scale = gr.Slider(minimum=0.1, maximum=1.5, value=0.6, step=0.1, label='感情变化')
 
107
  temperature=temperature):
108
  chatbot[-1] = (parse_text(input), parse_text(response))
109
 
110
+ yield chatbot, history, past_key_values, response
111
 
112
 
113
  def reset_user_input():
 
226
  max_length = gr.Slider(0, 32768, value=8192, step=1.0, label="Maximum length", interactive=True)
227
  top_p = gr.Slider(0, 1, value=0.8, step=0.01, label="Top P", interactive=True)
228
  temperature = gr.Slider(0, 1, value=0.95, step=0.01, label="Temperature", interactive=True)
229
+ response_lh = gr.Textbox(label="神里绫华的回答", visible=False)
230
 
231
  history = gr.State([])
232
  past_key_values = gr.State(None)
233
 
234
  submitBtn.click(predict, [user_input, chatbot, max_length, top_p, temperature, history, past_key_values],
235
+ [chatbot, history, past_key_values, response_lh], show_progress=True)
236
  submitBtn.click(reset_user_input, [], [user_input])
237
 
238
+ emptyBtn.click(reset_state, outputs=[chatbot, history, past_key_values, response_lh], show_progress=True)
239
 
240
 
241
  with gr.Row():
242
  with gr.Column():
243
+ text = response_lh
244
+ speaker = gr.Dropdown(choices=speakers, value=speakers[0], label='Speaker', visible=False)
 
245
  with gr.Row():
246
  sdp_ratio = gr.Slider(minimum=0, maximum=1, value=0.2, step=0.1, label='语调变化')
247
  noise_scale = gr.Slider(minimum=0.1, maximum=1.5, value=0.6, step=0.1, label='感情变化')