Add Radio

app.py

@@ -176,12 +176,14 @@ with demo:
                 - `pad_multiple = 64` for Tensorflow XLA models. Others do not pad (input prompts between 2 and 33 tokens).
                 """
             )
-            model_selector = gr.Dropdown(
-                choices=["DistilGPT2", "GPT2", "OPT-1.3B", "GPTJ-6B", "T5 Small", "T5 Base", "T5 Large", "T5 3B"],
-                value="T5 Small",
-                label="Model",
-                interactive=True,
-            )
+            with gr.Row():
+                model_selector = gr.Dropdown(
+                    choices=["DistilGPT2", "GPT2", "OPT-1.3B", "GPTJ-6B", "T5 Small", "T5 Base", "T5 Large", "T5 3B"],
+                    value="T5 Small",
+                    label="Model",
+                    interactive=True,
+                )
+                eager_enabler = gr.Radio(["Yes", "No"], value="Yes", label="Plot TF Eager Execution?")
             plot_fn = functools.partial(get_plot, generate_type="Greedy Search")
             plot = gr.Plot(value=plot_fn("T5 Small"))  # Show plot when the gradio app is initialized
             model_selector.change(fn=plot_fn, inputs=model_selector, outputs=plot)