Tonic commited on
Commit
670623f
·
1 Parent(s): ddffd15

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -73,7 +73,7 @@ examples = [
73
  starling_bot = StarlingBot()
74
 
75
  def gradio_starling(user_message, assistant_message, system_prompt, do_sample, temperature, max_new_tokens, top_p, repetition_penalty):
76
- response = starling_bot.predict(user_message, assistant_message, system_message, do_sample, temperature, max_new_tokens, top_p, repetition_penalty)
77
  return response
78
 
79
  with gr.Blocks(theme="ParityError/Anime") as demo:
@@ -98,7 +98,7 @@ with gr.Blocks(theme="ParityError/Anime") as demo:
98
 
99
  submit_button.click(
100
  gradio_starling,
101
- inputs=[user_message, assistant_message, system_message, do_sample, temperature, max_new_tokens, top_p, repetition_penalty],
102
  outputs=output_text
103
  )
104
 
 
73
  starling_bot = StarlingBot()
74
 
75
  def gradio_starling(user_message, assistant_message, system_prompt, do_sample, temperature, max_new_tokens, top_p, repetition_penalty):
76
+ response = starling_bot.predict(user_message, assistant_message, system_prompt, do_sample, temperature, max_new_tokens, top_p, repetition_penalty)
77
  return response
78
 
79
  with gr.Blocks(theme="ParityError/Anime") as demo:
 
98
 
99
  submit_button.click(
100
  gradio_starling,
101
+ inputs=[user_message, assistant_message, system_prompt, do_sample, temperature, max_new_tokens, top_p, repetition_penalty],
102
  outputs=output_text
103
  )
104