fiddle with plots

app.py

@@ -205,7 +205,7 @@ with demo:
                     interactive=True
                 )
             plot_fn = functools.partial(get_plot, generate_type="Greedy Search")
-            plot = gr.Image(value=plot_fn("T5 Small", "Yes"))  # Show plot when the gradio app is initialized
+            plot = gr.Image(value=plot_fn("T5 Small", "Yes"), shape=[1, 1])  # Show plot when the gradio app is initialized
             model_selector.change(fn=plot_fn, inputs=[model_selector, eager_enabler], outputs=plot)
             eager_enabler.change(fn=plot_fn, inputs=[model_selector, eager_enabler], outputs=plot)
         with gr.TabItem("Sample"):