mterris commited on
Commit
02ef7f2
·
1 Parent(s): f71c5d6
Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -200,10 +200,20 @@ generate_random_imgs_from_dataset_partial = partial(generate_random_imgs_from_da
200
  print(f"[Init] CUDA max allocated: {torch.cuda.max_memory_allocated() / 1024**2:.2f} MB")
201
  print(f"[Init] CUDA max reserved: {torch.cuda.max_memory_reserved() / 1024**2:.2f} MB")
202
 
203
- title = "Inverse problem playground" # displayed on gradio tab and in the gradio page
204
  with gr.Blocks(title=title, theme=gr.themes.Glass()) as interface:
205
  gr.Markdown("## " + title)
206
 
 
 
 
 
 
 
 
 
 
 
207
  ### USER-SPECIFIC VARIABLES
208
  dataset_placeholder = gr.State(get_dataset_on_DEVICE_STR("Natural"))
209
  available_physics_placeholder = gr.State(['Inpainting', 'SR', 'MotionBlur_medium', 'MotionBlur_hard',
 
200
  print(f"[Init] CUDA max allocated: {torch.cuda.max_memory_allocated() / 1024**2:.2f} MB")
201
  print(f"[Init] CUDA max reserved: {torch.cuda.max_memory_reserved() / 1024**2:.2f} MB")
202
 
203
+ title = "Reconstruct Anything Model Demo" # displayed on gradio tab and in the gradio page
204
  with gr.Blocks(title=title, theme=gr.themes.Glass()) as interface:
205
  gr.Markdown("## " + title)
206
 
207
+ gr.Markdown(
208
+ """
209
+ This demo showcases the performance of the **Reconstruct Anything Model (RAM)** across a variety of inverse problems on both natural and MRI images.
210
+ Select a dataset and a physics task below (e.g., inpainting, super-resolution, deblurring...).
211
+ Note: The parameters of the selected physics — such as noise levels, blur kernels, or inpainting masks — are randomly generated before reconstruction, leveraging the [deepinverse library](https://deepinv.github.io/deepinv/).
212
+
213
+ 📄 For more details on the method, check out our [paper on arXiv](https://arxiv.org/abs/2503.08915).
214
+ """
215
+ )
216
+
217
  ### USER-SPECIFIC VARIABLES
218
  dataset_placeholder = gr.State(get_dataset_on_DEVICE_STR("Natural"))
219
  available_physics_placeholder = gr.State(['Inpainting', 'SR', 'MotionBlur_medium', 'MotionBlur_hard',