Load automatically the right baseline and default physics when choosing a dataset

app.py

@@ -112,25 +112,23 @@ get_baseline_model_on_DEVICE_STR = partial(BaselineModel, device_str=DEVICE_STR)
 get_dataset_on_DEVICE_STR = partial(EvalDataset, device_str=DEVICE_STR)
 get_physics_on_DEVICE_STR = partial(PhysicsWithGenerator, device_str=DEVICE_STR)
 
-def get_physics(physics_name):
-    if physics_name == 'MRI':
-        baseline = get_baseline_model_on_DEVICE_STR('DPIR_MRI')
-    elif physics_name == 'CT':
-        baseline = get_baseline_model_on_DEVICE_STR('DPIR_CT')
-    else:
-        baseline = get_baseline_model_on_DEVICE_STR('DPIR')
-    return get_physics_on_DEVICE_STR(physics_name), baseline
-
 AVAILABLE_PHYSICS = PhysicsWithGenerator.all_physics
 def get_dataset(dataset_name):
     global AVAILABLE_PHYSICS
     if dataset_name == 'MRI':
         AVAILABLE_PHYSICS = ['MRI']
+        baseline_name = 'DPIR_MRI'
+        physics_name = 'MRI'
     elif dataset_name == 'CT':
         AVAILABLE_PHYSICS = ['CT']
+        baseline_name = 'DPIR_CT'
+        physics_name = 'CT'
     else:
         AVAILABLE_PHYSICS = ['MotionBlur_easy', 'MotionBlur_medium', 'MotionBlur_hard', 'GaussianBlur_easy', 'GaussianBlur_medium', 'GaussianBlur_hard']
-    return get_dataset_on_DEVICE_STR(dataset_name)
+        baseline_name = 'DPIR'
+        physics_name = 'MotionBlur_easy'
+    return get_dataset_on_DEVICE_STR(dataset_name), get_physics_on_DEVICE_STR(physics_name), get_baseline_model_on_DEVICE_STR(baseline_name)
+
 
 ### Gradio Blocks interface
 
@@ -212,10 +210,10 @@ with gr.Blocks(title=title, css=custom_css) as interface:
         ### Event listeners
         choose_dataset.change(fn=get_dataset,
                               inputs=choose_dataset,
-                              outputs=dataset_placeholder)
-        choose_physics.change(fn=get_physics,
+                              outputs=[dataset_placeholder, physics_placeholder, model_b_placeholder])
+        choose_physics.change(fn=get_physics_on_DEVICE_STR,
                               inputs=choose_physics,
-                              outputs=[physics_placeholder, model_b_placeholder])
+                              outputs=[physics_placeholder])
         update_button.click(fn=physics.update_and_display_params, inputs=[key_selector, value_text], outputs=physics_params)
         choose_metrics.change(fn=get_list_metrics_on_DEVICE_STR,
                               inputs=choose_metrics,