init

app.py

@@ -6,7 +6,7 @@ import torch
 import numpy as np
 from matplotlib.colors import LinearSegmentedColormap
 
-from app_config import CSS, TITLE, DESCRIPTION, BIBTEX, DEVICE
+from app_config import CSS, HEADER, FOOTER, DEVICE
 import sample_cond
 
 model = sample_cond.load_model()
@@ -30,7 +30,6 @@ def colorize_depth(depth, log_scale):
     return rgb
 
 
-@spaces.GPU
 @torch.no_grad()
 def generate_lidar(model, cond):
     img, pcd = sample_cond.sample(model, cond)
@@ -47,10 +46,7 @@ def load_camera(image):
 
 
 with gr.Blocks(css=CSS) as demo:
-    gr.Markdown(TITLE)
-    gr.Markdown(DESCRIPTION)
-    gr.Markdown("### Camera-to-LiDAR Demo")
-    # gr.Markdown("You can slide the output to compare the depth prediction with input image")
+    gr.Markdown(HEADER)
 
     with gr.Row():
         input_image = gr.Image(label="Input Image", type='numpy', elem_id='img-display-input')
@@ -76,7 +72,7 @@ with gr.Blocks(css=CSS) as demo:
     examples = gr.Examples(examples=example_files, inputs=[input_image], outputs=[output_image, raw_file],
                            fn=on_submit, cache_examples=True)
 
-    gr.Markdown(BIBTEX)
+    gr.Markdown(FOOTER)
 
 
 if __name__ == '__main__':

app_config.py

@@ -12,10 +12,16 @@ CSS = """
     }
 """
 DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
-TITLE = "# LiDAR Diffusion"
-DESCRIPTION = """Official demo for **LiDAR Diffusion: Towards Realistic Scene Generation with LiDAR Diffusion Models**.
-Please refer to our [paper](https://arxiv.org/abs/2404.00815), [project page](https://lidar-diffusion.github.io/), or [github](https://github.com/hancyran/LiDAR-Diffusion) for more details."""
-BIBTEX = r"""
+
+HEADER = '''
+# LiDAR Diffusion
+Official demo for **LiDAR Diffusion: Towards Realistic Scene Generation with LiDAR Diffusion Models**.
+Please refer to our [paper](https://arxiv.org/abs/2404.00815), [project page](https://lidar-diffusion.github.io/), or [github](https://github.com/hancyran/LiDAR-Diffusion) for more details.
+
+### Camera-to-LiDAR Demo
+'''
+
+FOOTER = r"""
 ```bibtex
 @inproceedings{ran2024towards,
     title={Towards Realistic Scene Generation with LiDAR Diffusion Models},