test

app.py

@@ -97,12 +97,13 @@ with demo:
                         """
                     )
                 # Show plot when the gradio app is initialized
-                plot = gr.Image(value=plot_fn("A100 (80GB)", "No"))
+                with gr.Column(scale=0.7):
+                    plot = gr.Image(value=plot_fn("A100 (80GB)", "No"))
             # 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)
             omit_offload.change(fn=plot_fn, inputs=plot_inputs, outputs=plot)
-        with gr.TabItem("OPT: Summarization"):
+        with gr.TabItem("OPT: Summ"):
             plot_fn = functools.partial(get_plot, "OPT: Summarization")
             with gr.Row():
                 with gr.Column(scale=0.3):
@@ -179,7 +180,7 @@ with demo:
             plot_inputs = [gpu_selector, omit_offload]
             gpu_selector.change(fn=plot_fn, inputs=plot_inputs, outputs=plot)
             omit_offload.change(fn=plot_fn, inputs=plot_inputs, outputs=plot)
-        with gr.TabItem("Flan-T5: Summarization"):
+        with gr.TabItem("Flan-T5: Summ"):
             plot_fn = functools.partial(get_plot, "Flan-T5: Summarization")
             with gr.Row():
                 with gr.Column(scale=0.3):