Update app.py

app.py

@@ -302,6 +302,24 @@ with gr.Blocks() as demo:
             strategy_selected = gr.Radio (["Sampling", "Beam Search", "Diversity Beam Search","Contrastive"], label="Search strategy", value = "Sampling", interactive=True)
             
         
+        with gr.Column(scale=1):
+                
+            text = gr.Textbox(
+            label="Prompt",
+            autoscroll=True,
+            value="It's a rainy day today"
+            )
+
+            out_markdown = gr.Textbox(label="Output", autoscroll=True)
+
+            button = gr.Button("Generate")  
+              
+            cleared = gr.Button ("Clear")
+            cleared.click (fn=clear, inputs=[], outputs=[out_markdown])
+
+        
+        
+        
         with gr.Column (scale=0, min_width=250) as Beam_Params:
             n_steps = gr.Slider(
                 label="Number of steps/tokens", minimum=1, maximum=100, step=1, value=20 
@@ -357,15 +375,6 @@ with gr.Blocks() as demo:
                 label="Repetition penalty", minimum=0, maximum=3, step=1, value=float(0), info="Prior context based penalty for unique text"
             )
 
-        with gr.Column(scale=2):
-            
-            text = gr.Textbox(
-            label="Prompt",
-            autoscroll=True,
-            value="It's a rainy day today"
-            )
-
-            out_markdown = gr.Textbox(label="Output", autoscroll=True)
     
 
 #----------ON SELECTING/CHANGING: RETURN SEEQUENCES/NO OF BEAMS/BEAM GROUPS/TEMPERATURE--------
@@ -393,19 +402,12 @@ with gr.Blocks() as demo:
 
     #-------------GENERATE BUTTON-------------------
     
-    with gr.Row():
-        with gr.Column (scale=0, min_width=200):
-    
-            button = gr.Button("Generate")  
-            
-            button.click(
-                fn = generate,
-                inputs=[text, n_steps, n_beams, beam_groups, diversity_penalty, length_penalty, num_return_sequences, temperature, no_repeat_ngram_size, repetition_penalty, early_stopping, beam_temperature, top_p, top_k,penalty_alpha,top_p_box,top_k_box,strategy_selected,model_selected],
-                outputs=[out_markdown]
-            )
-        with gr.Column (scale=0, min_width=200):    
-            cleared = gr.Button ("Clear")
-            cleared.click (fn=clear, inputs=[], outputs=[out_markdown])
+    button.click(
+        fn = generate,
+        inputs=[text, n_steps, n_beams, beam_groups, diversity_penalty, length_penalty, num_return_sequences, temperature, no_repeat_ngram_size, repetition_penalty, early_stopping, beam_temperature, top_p, top_k,penalty_alpha,top_p_box,top_k_box,strategy_selected,model_selected],
+        outputs=[out_markdown]
+    )
+
 
     with gr.Row():