changed dropdown to slider (default value removed)
Browse files
app.py
CHANGED
@@ -139,7 +139,7 @@ classify_interface = gr.Interface(
|
|
139 |
attention_interface = gr.Interface(
|
140 |
fn=plot_attention,
|
141 |
inputs=[gr.Image(type="pil", label="Image"),
|
142 |
-
gr.Slider(1, 12,
|
143 |
outputs=gr.Gallery(value=attention_maps, label="Attention Maps").style(grid=(3, 4)),
|
144 |
examples=example_list,
|
145 |
title=title_attention,
|
|
|
139 |
attention_interface = gr.Interface(
|
140 |
fn=plot_attention,
|
141 |
inputs=[gr.Image(type="pil", label="Image"),
|
142 |
+
gr.Slider(1, 12, step=1, label="Transformer Encoder Layer")],
|
143 |
outputs=gr.Gallery(value=attention_maps, label="Attention Maps").style(grid=(3, 4)),
|
144 |
examples=example_list,
|
145 |
title=title_attention,
|