Spaces:

dar-tau
/

minimal-attn-maps

Sleeping

dar-tau commited on Mar 9

Commit

dae383f

•

1 Parent(s): c5fa8a7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -49,7 +49,7 @@ with demo:
         sentence_dropdown = gr.Dropdown(label="Sentence",
                                         choices=[x.split('</s> ')[1] for x in dataset['text']],
                                value=0, min_width=500, type='index')
-        vis_dropdown = gr.Dropdown(label="Visualization", choices=[x.value for x in VisType] + ['Layer #' + i for i in range(num_layers)],
                                    min_width=150, value=VisType.SUM, type='value')
         btn = gr.Button("Run", min_width=30)
     output = gr.Plot(label="Plot", container=True)

         sentence_dropdown = gr.Dropdown(label="Sentence",
                                         choices=[x.split('</s> ')[1] for x in dataset['text']],
                                value=0, min_width=500, type='index')
+        vis_dropdown = gr.Dropdown(label="Visualization", choices=[x.value for x in VisType] + [f'Layer #{i}' for i in range(num_layers)],
                                    min_width=150, value=VisType.SUM, type='value')
         btn = gr.Button("Run", min_width=30)
     output = gr.Plot(label="Plot", container=True)