aarbelle commited on
Commit
03be69d
1 Parent(s): d9eddff

add min len to slider

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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, 1000)
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():