mterris commited on
Commit
c7ae131
·
1 Parent(s): 88d3587
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -192,11 +192,11 @@ with gr.Blocks(title=title, theme=gr.themes.Glass()) as interface:
192
 
193
  ### USER-SPECIFIC VARIABLES
194
  dataset_placeholder = gr.State(get_dataset_on_DEVICE_STR("Natural"))
195
- available_physics_placeholder = gr.State(['MotionBlur_easy', 'MotionBlur_medium', 'MotionBlur_hard',
196
  'GaussianBlur_easy', 'GaussianBlur_medium', 'GaussianBlur_hard'])
197
  # Issue giving directly a `torch.nn.module` to `gr.State(...)` since it has __call__ method
198
  # Solution: using lambda expression
199
- physics_placeholder = gr.State(lambda: get_physics_on_DEVICE_STR("MotionBlur_easy"))
200
  model_b_placeholder = gr.State(lambda: get_baseline_model_on_DEVICE_STR("DPIR"))
201
 
202
  print(f"[Render] CUDA max allocated: {torch.cuda.max_memory_allocated() / 1024**2:.2f} MB")
 
192
 
193
  ### USER-SPECIFIC VARIABLES
194
  dataset_placeholder = gr.State(get_dataset_on_DEVICE_STR("Natural"))
195
+ available_physics_placeholder = gr.State(['MotionBlur_medium', 'MotionBlur_hard',
196
  'GaussianBlur_easy', 'GaussianBlur_medium', 'GaussianBlur_hard'])
197
  # Issue giving directly a `torch.nn.module` to `gr.State(...)` since it has __call__ method
198
  # Solution: using lambda expression
199
+ physics_placeholder = gr.State(lambda: get_physics_on_DEVICE_STR("MotionBlur_medium"))
200
  model_b_placeholder = gr.State(lambda: get_baseline_model_on_DEVICE_STR("DPIR"))
201
 
202
  print(f"[Render] CUDA max allocated: {torch.cuda.max_memory_allocated() / 1024**2:.2f} MB")