spacing

app.py

@@ -168,11 +168,11 @@ generate_imgs_from_dataset_partial = partial(generate_imgs_from_dataset, model=r
 generate_random_imgs_from_dataset_partial = partial(generate_random_imgs_from_dataset, model=ram_model, metrics=psnr)
 
 
+### Gradio Blocks interface
+
 print(f"[Init] CUDA max allocated: {torch.cuda.max_memory_allocated() / 1024**2:.2f} MB")
 print(f"[Init] CUDA max reserved: {torch.cuda.max_memory_reserved() / 1024**2:.2f} MB")
 
-### Gradio Blocks interface
-
 title = "Inverse problem playground"  # displayed on gradio tab and in the gradio page
 with gr.Blocks(title=title, theme=gr.themes.Glass()) as interface:
     gr.Markdown("## " + title)
@@ -188,6 +188,7 @@ with gr.Blocks(title=title, theme=gr.themes.Glass()) as interface:
 
     print(f"[Render] CUDA max allocated: {torch.cuda.max_memory_allocated() / 1024**2:.2f} MB")
     print(f"[Render] CUDA max reserved: {torch.cuda.max_memory_reserved() / 1024**2:.2f} MB")
+
     @gr.render(inputs=[dataset_placeholder, physics_placeholder, available_physics_placeholder])
     def dynamic_layout(dataset, physics, available_physics):
         ### LAYOUT