Update app.py

app.py

@@ -53,9 +53,11 @@ with demo:
         vis_dropdown = gr.Dropdown(label="Visualization", choices=[x.value for x in VisType] + [f'Layer #{i}' for i in range(num_layers)], 
                                    min_width=70, value=VisType.SUM, type='value')
         btn = gr.Button("Run", min_width=30)
+    
+    vis_format_checkbox = gr.Radio(['Plot', 'Graph'])
+        
     output = gr.Plot(label="Plot", container=True)
     metrics = gr.Label("Metrics")
-    vis_format_checkbox = gr.Radio(['Plot', 'Graph'])
     btn.click(analyze_sentence, [sentence_dropdown, vis_dropdown, vis_format_checkbox], [output, metrics])