kz209 commited on
Commit
453ac93
β€’
1 Parent(s): 4a6e70a
Files changed (1) hide show
  1. pages/summarization_playground.py +3 -3
pages/summarization_playground.py CHANGED
@@ -143,14 +143,14 @@ def create_summarization_interface():
143
  submit_button = gr.Button("✨ Submit ✨")
144
 
145
  with gr.Row():
146
- with gr.Row():
147
  with gr.Column():
148
  gr.Markdown("<div style='border: 4px solid white; padding: 2px; border-radius: 5px;width:130px;padding-bottom: 10px;'><b><h3>Parameters πŸ“ˆ</h3></center></b></div>")
149
  with gr.Column():
150
  temperature = gr.Textbox(label="Temperature",elem_classes="parameter-text", value=0.0)
151
  max_new_tokens = gr.Textbox(label="Max New Tokens",elem_classes="parameter-text", value=500)
152
- do_sample = gr.Dropdown(['Default','None'],label="Do Sample",elem_classes="parameter-text", value=True)
153
- output = gr.Markdown(line_breaks=True)
154
 
155
  example_dropdown.change(update_input, inputs=[example_dropdown], outputs=[input_text])
156
  submit_button.click(process_input, inputs=[input_text,model_dropdown,Template_text,temperature,max_new_tokens,do_sample], outputs=[output])
 
143
  submit_button = gr.Button("✨ Submit ✨")
144
 
145
  with gr.Row():
146
+ with gr.Row(scale=1):
147
  with gr.Column():
148
  gr.Markdown("<div style='border: 4px solid white; padding: 2px; border-radius: 5px;width:130px;padding-bottom: 10px;'><b><h3>Parameters πŸ“ˆ</h3></center></b></div>")
149
  with gr.Column():
150
  temperature = gr.Textbox(label="Temperature",elem_classes="parameter-text", value=0.0)
151
  max_new_tokens = gr.Textbox(label="Max New Tokens",elem_classes="parameter-text", value=500)
152
+ do_sample = gr.Dropdown([True,False],label="Do Sample",elem_classes="parameter-text", value=True)
153
+ output = gr.Markdown(line_breaks=True, scale=2)
154
 
155
  example_dropdown.change(update_input, inputs=[example_dropdown], outputs=[input_text])
156
  submit_button.click(process_input, inputs=[input_text,model_dropdown,Template_text,temperature,max_new_tokens,do_sample], outputs=[output])