sneha commited on
Commit ·
ba48df9
1
Parent(s): f038435
change default
Browse files
app.py
CHANGED
|
@@ -95,7 +95,7 @@ def run_attn(input_img, model="vc1-base",fusion="min"):
|
|
| 95 |
return attn_img, fig
|
| 96 |
|
| 97 |
model_type = gr.Dropdown(
|
| 98 |
-
["vc1-base", "vc1-large"], label="Model Size", value="vc1-
|
| 99 |
input_img = gr.Image(shape=(250,250))
|
| 100 |
input_button = gr.Radio(["min", "max", "mean"], value="min",label="Attention Head Fusion", info="How to combine the last layer attention across all 12 heads of the transformer.")
|
| 101 |
output_img = gr.Image(shape=(250,250))
|
|
|
|
| 95 |
return attn_img, fig
|
| 96 |
|
| 97 |
model_type = gr.Dropdown(
|
| 98 |
+
["vc1-base", "vc1-large"], label="Model Size", value="vc1-base")
|
| 99 |
input_img = gr.Image(shape=(250,250))
|
| 100 |
input_button = gr.Radio(["min", "max", "mean"], value="min",label="Attention Head Fusion", info="How to combine the last layer attention across all 12 heads of the transformer.")
|
| 101 |
output_img = gr.Image(shape=(250,250))
|