Spaces:
Sleeping
Sleeping
add min len to slider
Browse files
app.py
CHANGED
@@ -29,9 +29,11 @@ else:
|
|
29 |
src_data_dict[search_domain] = (src_data,src_nn_fit)
|
30 |
|
31 |
dst_data_dict = {}
|
|
|
32 |
for d in domains:
|
33 |
with open(os.path.join(feat_dir, f'src_{d}_{shots}.pkl'), 'rb') as fp:
|
34 |
dst_data_dict[d] = pickle.load(fp)
|
|
|
35 |
|
36 |
def query(query_index, query_domain):
|
37 |
dst_data = dst_data_dict[query_domain]
|
@@ -57,7 +59,7 @@ with demo:
|
|
57 |
gr.Markdown('## Select Query Domain: ')
|
58 |
domain_drop = gr.Dropdown(domains)
|
59 |
# domain_select_button = gr.Button("Select Domain")
|
60 |
-
slider = gr.Slider(0,
|
61 |
image_button = gr.Button("Run")
|
62 |
|
63 |
with gr.Row():
|
|
|
29 |
src_data_dict[search_domain] = (src_data,src_nn_fit)
|
30 |
|
31 |
dst_data_dict = {}
|
32 |
+
min_len = 1e10
|
33 |
for d in domains:
|
34 |
with open(os.path.join(feat_dir, f'src_{d}_{shots}.pkl'), 'rb') as fp:
|
35 |
dst_data_dict[d] = pickle.load(fp)
|
36 |
+
min_len = min(min_len,(dst_data_dict[d][0]))
|
37 |
|
38 |
def query(query_index, query_domain):
|
39 |
dst_data = dst_data_dict[query_domain]
|
|
|
59 |
gr.Markdown('## Select Query Domain: ')
|
60 |
domain_drop = gr.Dropdown(domains)
|
61 |
# domain_select_button = gr.Button("Select Domain")
|
62 |
+
slider = gr.Slider(0, min_len)
|
63 |
image_button = gr.Button("Run")
|
64 |
|
65 |
with gr.Row():
|