🚧

Signed-off-by: peter szemraj <[email protected]>

app.py

@@ -20,10 +20,10 @@ def generate_text(
     num_beams=4,
     no_repeat_ngram_size=2,
     length_penalty=1.0,
+    num_beam_groups=2,
     # perma params (not set by user)
     repetition_penalty=3.5,
     abs_max_length=512,
-    num_beam_groups=2,
     verbose=False,
 ):
     """
@@ -215,19 +215,22 @@ if __name__ == "__main__":
                     "Load Model",
                     variant="secondary",
                 )
-
+                no_repeat_ngram_size = gr.Radio(
+                                    choices=[1, 2, 3, 4],
+                                    label="no repeat ngram size",
+                                    value=2,
+                                )
             with gr.Row():
                 num_beams = gr.Radio(
                     choices=[4, 8, 12, 16],
                     label="Number of Beams",
                     value=4,
                 )
-            with gr.Row():
 
-                no_repeat_ngram_size = gr.Radio(
-                    choices=[1, 2, 3, 4],
-                    label="no repeat ngram size",
-                    value=2,
+                num_beam_groups = gr.Radio(
+                    choices=[1, 2],
+                    label="Number of Beam Groups",
+                    value=1,
                 )
                 length_penalty = gr.Slider(
                     minimum=0.5,
@@ -262,8 +265,9 @@ if __name__ == "__main__":
                 num_beams,
                 no_repeat_ngram_size,
                 length_penalty,
+                num_beam_groups,
             ],
-            outputs=[generated_email, email_template],
+            outputs=[email_template],
         )
 
         load_model_button.click(