Spaces:
Running
on
T4
Running
on
T4
txt
Browse files
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 = "
|
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',
|