mterris commited on
Commit
250bf66
·
1 Parent(s): abfbafe
Files changed (1) hide show
  1. app.py +9 -9
app.py CHANGED
@@ -67,7 +67,7 @@ def generate_random_imgs_from_dataset(dataset: EvalDataset,
67
  metrics: List[Metric]):
68
  idx = random.randint(0, len(dataset)-1)
69
  x, y, out, out_baseline, saved_params_str, metrics_y, metrics_out, metrics_out_baseline = generate_imgs_from_dataset(
70
- dataset, idx, physics, True, baseline, model, metrics
71
  )
72
  return idx, x, y, out, out_baseline, saved_params_str, metrics_y, metrics_out, metrics_out_baseline
73
 
@@ -263,13 +263,13 @@ with gr.Blocks(title=title, theme=gr.themes.Glass()) as interface:
263
  value=dataset.name)
264
  idx_slider = gr.Slider(minimum=0, maximum=len(dataset) - 1, step=1, label="Sample index",
265
  key='idx_slider')
 
 
 
 
 
 
266
  with gr.Column(scale=1):
267
- with gr.Row():
268
- key_selector = gr.Dropdown(choices=list(physics.saved_params["updatable_params"].keys()),
269
- label="Updatable Key")
270
- value_text = gr.Textbox(label="Update Value")
271
- update_button = gr.Button("Manually update parameter value", size='md')
272
- with gr.Column(scale=2):
273
  physics_params = gr.Textbox(label="Physics parameters",
274
  lines=5,
275
  value=physics.display_saved_params())
@@ -282,8 +282,8 @@ with gr.Blocks(title=title, theme=gr.themes.Glass()) as interface:
282
  choose_physics.change(fn=get_physics_on_DEVICE_STR,
283
  inputs=choose_physics,
284
  outputs=[physics_placeholder])
285
- update_button.click(fn=physics.update_and_display_params,
286
- inputs=[key_selector, value_text], outputs=physics_params)
287
  run_button.click(fn=generate_imgs_from_user_partial,
288
  inputs=[gt_img,
289
  physics_placeholder,
 
67
  metrics: List[Metric]):
68
  idx = random.randint(0, len(dataset)-1)
69
  x, y, out, out_baseline, saved_params_str, metrics_y, metrics_out, metrics_out_baseline = generate_imgs_from_dataset(
70
+ dataset, idx, physics, baseline, model, metrics
71
  )
72
  return idx, x, y, out, out_baseline, saved_params_str, metrics_y, metrics_out, metrics_out_baseline
73
 
 
263
  value=dataset.name)
264
  idx_slider = gr.Slider(minimum=0, maximum=len(dataset) - 1, step=1, label="Sample index",
265
  key='idx_slider')
266
+ # with gr.Column(scale=1):
267
+ # with gr.Row():
268
+ # key_selector = gr.Dropdown(choices=list(physics.saved_params["updatable_params"].keys()),
269
+ # label="Updatable Key")
270
+ # value_text = gr.Textbox(label="Update Value")
271
+ # update_button = gr.Button("Manually update parameter value", size='md')
272
  with gr.Column(scale=1):
 
 
 
 
 
 
273
  physics_params = gr.Textbox(label="Physics parameters",
274
  lines=5,
275
  value=physics.display_saved_params())
 
282
  choose_physics.change(fn=get_physics_on_DEVICE_STR,
283
  inputs=choose_physics,
284
  outputs=[physics_placeholder])
285
+ # update_button.click(fn=physics.update_and_display_params,
286
+ # inputs=[key_selector, value_text], outputs=physics_params)
287
  run_button.click(fn=generate_imgs_from_user_partial,
288
  inputs=[gt_img,
289
  physics_placeholder,