Spaces:
Sleeping
Sleeping
rm easy
Browse files
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(['
|
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("
|
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")
|