Spaces:
Running
on
Zero
Running
on
Zero
update defaults
Browse files
app.py
CHANGED
@@ -136,11 +136,13 @@ def get_random_path(length=10):
|
|
136 |
return path
|
137 |
|
138 |
default_images = ['./images/image_0.jpg', './images/image_1.jpg', './images/image_2.jpg', './images/image_3.jpg', './images/image_5.jpg']
|
139 |
-
default_outputs = ['./images/ncut_0.jpg', './images/ncut_1.jpg', './images/ncut_2.jpg', './images/ncut_3.jpg', './images/ncut_5.jpg']
|
|
|
140 |
default_outputs_independent = ['./images/ncut_0_independent.jpg', './images/ncut_1_independent.jpg', './images/ncut_2_independent.jpg', './images/ncut_3_independent.jpg', './images/ncut_5_independent.jpg']
|
141 |
|
142 |
downscaled_images = ['./images/image_0_small.jpg', './images/image_1_small.jpg', './images/image_2_small.jpg', './images/image_3_small.jpg', './images/image_5_small.jpg']
|
143 |
-
downscaled_outputs = ['./images/ncut_0_small.jpg', './images/ncut_1_small.jpg', './images/ncut_2_small.jpg', './images/ncut_3_small.jpg', './images/ncut_5_small.jpg']
|
|
|
144 |
|
145 |
example_items = downscaled_images[:3] + downscaled_outputs[:3]
|
146 |
|
@@ -481,7 +483,7 @@ def make_parameters_section():
|
|
481 |
"SAM2(sam2_hiera_b+)",
|
482 |
"SAM2(sam2_hiera_t)",
|
483 |
]
|
484 |
-
model_dropdown = gr.Dropdown(model_names, label="Backbone", value="
|
485 |
layer_slider = gr.Slider(0, 11, step=1, label="Backbone: Layer index", value=11, elem_id="layer")
|
486 |
node_type_dropdown = gr.Dropdown(["attn: attention output", "mlp: mlp output", "block: sum of residual"], label="Backbone: Layer type", value="block: sum of residual", elem_id="node_type", info="which feature to take from each layer?")
|
487 |
num_eig_slider = gr.Slider(1, 1000, step=1, label="NCUT: Number of eigenvectors", value=100, elem_id="num_eig", info='increase for more clusters')
|
|
|
136 |
return path
|
137 |
|
138 |
default_images = ['./images/image_0.jpg', './images/image_1.jpg', './images/image_2.jpg', './images/image_3.jpg', './images/image_5.jpg']
|
139 |
+
# default_outputs = ['./images/ncut_0.jpg', './images/ncut_1.jpg', './images/ncut_2.jpg', './images/ncut_3.jpg', './images/ncut_5.jpg']
|
140 |
+
default_outputs = ['./images/image-1.webp', './images/image-2.webp', './images/image-3.webp', './images/image-4.webp', './images/image-5.webp']
|
141 |
default_outputs_independent = ['./images/ncut_0_independent.jpg', './images/ncut_1_independent.jpg', './images/ncut_2_independent.jpg', './images/ncut_3_independent.jpg', './images/ncut_5_independent.jpg']
|
142 |
|
143 |
downscaled_images = ['./images/image_0_small.jpg', './images/image_1_small.jpg', './images/image_2_small.jpg', './images/image_3_small.jpg', './images/image_5_small.jpg']
|
144 |
+
# downscaled_outputs = ['./images/ncut_0_small.jpg', './images/ncut_1_small.jpg', './images/ncut_2_small.jpg', './images/ncut_3_small.jpg', './images/ncut_5_small.jpg']
|
145 |
+
downscaled_outputs = default_outputs
|
146 |
|
147 |
example_items = downscaled_images[:3] + downscaled_outputs[:3]
|
148 |
|
|
|
483 |
"SAM2(sam2_hiera_b+)",
|
484 |
"SAM2(sam2_hiera_t)",
|
485 |
]
|
486 |
+
model_dropdown = gr.Dropdown(model_names, label="Backbone", value="SAM2(sam2_hiera_t)", elem_id="model_name")
|
487 |
layer_slider = gr.Slider(0, 11, step=1, label="Backbone: Layer index", value=11, elem_id="layer")
|
488 |
node_type_dropdown = gr.Dropdown(["attn: attention output", "mlp: mlp output", "block: sum of residual"], label="Backbone: Layer type", value="block: sum of residual", elem_id="node_type", info="which feature to take from each layer?")
|
489 |
num_eig_slider = gr.Slider(1, 1000, step=1, label="NCUT: Number of eigenvectors", value=100, elem_id="num_eig", info='increase for more clusters')
|