updated colour scheme

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from src.utils import *
 
 if __name__ == '__main__':
-    theme = gr.themes.Soft(primary_hue="red", secondary_hue="red", font=['arial'])
+    theme = gr.themes.Soft(primary_hue="emerald", secondary_hue="stone", font=[gr.themes.GoogleFont("Source Sans 3", weights=(400, 600)),'arial'])
     
     with gr.Blocks(theme=theme) as demo:
         with gr.Column(elem_classes="header"):
@@ -10,7 +10,6 @@ if __name__ == '__main__':
             gr.Markdown("### Paul Borne–Pons, Mikolaj Czerkawski, Rosalie Martin, Romain Rouffet")
             gr.Markdown('[[GitHub](https://github.com/PaulBorneP/MESA)] [[Model](https://huggingface.co/NewtNewt/MESA)] [[Dataset](https://huggingface.co/datasets/Major-TOM/Core-DEM)]')
             
-        # Abstract Section
         with gr.Column(elem_classes="abstract"):
             gr.Markdown("MESA is a novel generative model based on latent denoising diffusion capable of generating 2.5D representations of terrain based on the text prompt conditioning supplied via natural language. The model produces two co-registered modalities of optical and depth maps.") # Replace with your abstract text
             gr.Markdown("This is a test version of the demo app. Please be aware that MESA supports primarily complex, mountainous terrains as opposed to flat land")
@@ -37,4 +36,4 @@ if __name__ == '__main__':
             outputs=model_output,
         )
         
-        demo.queue().launch()
+        demo.queue().launch()

src/build_pipe.py

@@ -1,5 +1,4 @@
 from .pipeline_terrain import TerrainDiffusionPipeline
-#import models
 from huggingface_hub import hf_hub_download, snapshot_download
 import os
 import torch