Update app.py

app.py

@@ -8,7 +8,7 @@ from diffusers import DiffusionPipeline
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-pipe = DiffusionPipeline.from_pretrained("shuttleai/shuttle-3-diffusion", torch_dtype=dtype).to(device)
+pipe = DiffusionPipeline.from_pretrained("shuttleai/shuttle-3.1-aesthetic", torch_dtype=dtype).to(device)
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
@@ -44,8 +44,8 @@ css="""
 with gr.Blocks(css=css) as demo:
     
     with gr.Column(elem_id="col-container"):
-        gr.Markdown(f"""# Shuttle 3 Diffusion
-Shuttle 3 Diffusion is a text-to-image AI model designed to create detailed and diverse images from textual prompts in just 4 steps. It offers enhanced performance in image quality, typography, understanding complex prompts, and resource efficiency.
+        gr.Markdown(f"""# Shuttle 3.1 Aesthetic
+Shuttle 3.1 Aesthetic is a text-to-image AI model designed to create aesthetic, detailed and diverse images from textual prompts in just 4 steps. It offers enhanced performance in image quality, typography, understanding complex prompts, and resource efficiency.
         """)
         
         with gr.Row():