Fix figure not show bug

app.py

@@ -59,8 +59,6 @@ def plot_figs(fig_num, elev, azim):
     ax.yaxis.set_ticklabels([])
     ax.zaxis.set_ticklabels([])
 
-    plt.savefig(f"{fig_num}.png")
-
     return fig
 
 def make_plot(plot_type):
@@ -72,7 +70,7 @@ def make_plot(plot_type):
         elev = 30
         azim = 20
         fig_num = 2
-    plot_figs(fig_num, elev, azim)
+    return plot_figs(fig_num, elev, azim)
 
 title = "Principal components analysis (PCA)"
 with gr.Blocks(title=title) as demo:
@@ -84,7 +82,8 @@ with gr.Blocks(title=title) as demo:
     button = gr.Radio(label="Plot type",
                       choices=['Very flat direction', 'Not flat direction'], value='Very flat direction')
     plot = gr.Plot(label="Plot")
-    button.change(make_plot, inputs=button, outputs=[plot])
-    demo.load(make_plot, inputs=[button], outputs=[plot])
+    button.change(make_plot, inputs=button, outputs=plot)
+    demo.load(make_plot, inputs=button, outputs=plot)
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()