update

pages/summarization_playground.py

@@ -117,7 +117,7 @@ def generate_answer(sources, model_name, prompt, temperature, max_new_tokens, do
 def process_input(input_text, model_selection, prompt, temperature, max_new_tokens, do_sample):
     if input_text:
         logging.info("Start generation")
-        response = generate_answer(input_text, model_selection, prompt)
+        response = generate_answer(input_text, model_selection, prompt, temperature, max_new_tokens, do_sample)
         return f"## Original Dialogue:\n\n{input_text}\n\n## Summarization:\n\n{response}"
     else:
         return "Please fill the input to generate outputs."
@@ -149,7 +149,8 @@ def create_summarization_interface():
                     temperature = gr.Textbox(label="Temperature",elem_classes="parameter-text", value=0.0)
                     max_new_tokens = gr.Textbox(label="Max New Tokens",elem_classes="parameter-text", value=500)
                     do_sample = gr.Dropdown([True,False],label="Do Sample",elem_classes="parameter-text", value=True)
-            output = gr.Markdown(line_breaks=True)
+            with gr.Column(scale=3):
+                output = gr.Markdown(line_breaks=True)
 
         example_dropdown.change(update_input, inputs=[example_dropdown], outputs=[input_text])
         submit_button.click(process_input, inputs=[input_text,model_dropdown,Template_text,temperature,max_new_tokens,do_sample], outputs=[output])