Update app.py

app.py

@@ -2,7 +2,7 @@ import torch
 from diffusers import StableDiffusionPipeline, DDIMScheduler
 import gradio as gr
 
-pipe = StableDiffusionPipeline.from_pretrained("MVRL/GeoSynth")
+pipe = StableDiffusionPipeline.from_pretrained("MVRL/GeoSynth", torch_dtype=torch.float16)
 # scheduler = DDIMScheduler.from_pretrained("stabilityai/stable-diffusion-2-1-base")
 # pipe.scheduler = scheduler
 
@@ -23,7 +23,12 @@ def process(prompt, n_prompt, num_samples, image_resolution, ddim_steps, scale,
 block = gr.Blocks().queue()
 with block:
     with gr.Row():
-        gr.Markdown("## Control Stable Diffusion with Depth Maps")
+        gr.Markdown(
+            """
+            # GeoSynth: Contextually-Aware High-Resolution Satellite Image Synthesis
+            Srikumar Sastry*, Subash Khanal, Aayush Dhakal, Nathan Jacobs (*Corresponding Author)<br>
+            """
+        )
     with gr.Row():
         with gr.Column():
             prompt = gr.Textbox(label="Prompt")