dual col options

app.py

@@ -78,25 +78,27 @@ with demo:
         with gr.TabItem("OPT: Open"):
             plot_fn = functools.partial(get_plot, "OPT: Open Text Generation")
             with gr.Row():
-                gpu_selector = gpu_selector_fn(["3090", "T4", "T4 *2", "A100 (80GB)"])
-                omit_offload = omit_offload_fn()
-                # Show plot when the gradio app is initialized
-                plot = gr.Image(value=plot_fn("A100 (80GB)", "No"))
-                gr.Markdown(
-                    """
-                    ### Assistant Model
-                    - `facebook/opt-125m`
-
-                    ### Model Names:
-                    - 1.3B: `facebook/opt-1.3b`
-                    - 6.7B: `facebook/opt-6.7b`
-                    - 30B: `facebook/opt-30b`
-                    - 66B: `facebook/opt-66b`
-
-                    ### Dataset used as input prompt:
-                    - C4 (en, validation set)
-                    """
-                )
+                with gr.Column():
+                    gpu_selector = gpu_selector_fn(["3090", "T4", "T4 *2", "A100 (80GB)"])
+                with gr.Column():
+                    omit_offload = omit_offload_fn()
+            # Show plot when the gradio app is initialized
+            plot = gr.Image(value=plot_fn("A100 (80GB)", "No"))
+            gr.Markdown(
+                """
+                ### Assistant Model
+                - `facebook/opt-125m`
+
+                ### Model Names:
+                - 1.3B: `facebook/opt-1.3b`
+                - 6.7B: `facebook/opt-6.7b`
+                - 30B: `facebook/opt-30b`
+                - 66B: `facebook/opt-66b`
+
+                ### Dataset used as input prompt:
+                - C4 (en, validation set)
+                """
+            )
             # Update plot when any of the inputs change
             plot_inputs = [gpu_selector, omit_offload]
             gpu_selector.change(fn=plot_fn, inputs=plot_inputs, outputs=plot)