Update app.py

app.py

@@ -31,7 +31,7 @@ pipeline.load_lora_weights(lora_weights_path)
 pipeline = pipeline.to(device)
 
 MAX_SEED = np.iinfo(np.int32).max
-MAX_IMAGE_SIZE = 768  # Reduce max image size to fit within memory constraints
+MAX_IMAGE_SIZE = 1024  # Reduce max image size to fit within memory constraints
 
 @spaces.GPU
 def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
@@ -104,6 +104,28 @@ body {
     border-radius: 8px;
     margin-top: 20px;
 }
+
+.color-block {
+    height: 100px;
+    width: 100%;
+    margin-bottom: 10px;
+}
+
+.color-block-pink {
+    background-color: #ff3f6c;
+}
+
+.color-block-blue {
+    background-color: #3498db;
+}
+
+.color-block-green {
+    background-color: #2ecc71;
+}
+
+.color-block-yellow {
+    background-color: #f1c40f;
+}
 """
 
 if torch.cuda.is_available():
@@ -118,6 +140,11 @@ with gr.Blocks(css=css) as demo:
         Currently running on {power_device}.
         """)
         
+        gr.HTML("<div class='color-block color-block-pink'></div>")
+        gr.HTML("<div class='color-block color-block-blue'></div>")
+        gr.HTML("<div class='color-block color-block-green'></div>")
+        gr.HTML("<div class='color-block color-block-yellow'></div>")
+
         with gr.Row():
             prompt = gr.Textbox(
                 label="Prompt",