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")