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))
|