sneha commited on
Commit
ba48df9
1 Parent(s): f038435

change default

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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-large")
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))