Update app.py

app.py

@@ -89,6 +89,56 @@ def main(image_path, b_lora_trained_folder, instance_prompt):
 
 css = """
 #col-container {max-width: 780px; margin-left: auto; margin-right: auto;}
+
+div#warning-ready {
+    background-color: #ecfdf5;
+    padding: 0 10px 5px;
+    margin: 20px 0;
+}
+div#warning-ready > .gr-prose > h2, div#warning-ready > .gr-prose > p {
+    color: #057857!important;
+}
+div#warning-duplicate {
+    background-color: #ebf5ff;
+    padding: 0 10px 5px;
+    margin: 20px 0;
+}
+div#warning-duplicate > .gr-prose > h2, div#warning-duplicate > .gr-prose > p {
+    color: #0f4592!important;
+}
+div#warning-duplicate strong {
+    color: #0f4592;
+}
+p.actions {
+    display: flex;
+    align-items: center;
+    margin: 20px 0;
+}
+div#warning-duplicate .actions a {
+    display: inline-block;
+    margin-right: 10px;
+}
+div#warning-setgpu {
+    background-color: #fff4eb;
+    padding: 0 10px 5px;
+    margin: 20px 0;
+}
+div#warning-setgpu > .gr-prose > h2, div#warning-setgpu > .gr-prose > p {
+    color: #92220f!important;
+}
+div#warning-setgpu a, div#warning-setgpu b {
+    color: #91230f;
+}
+div#warning-setgpu p.actions > a {
+    display: inline-block;
+    background: #1f1f23;
+    border-radius: 40px;
+    padding: 6px 24px;
+    color: antiquewhite;
+    text-decoration: none;
+    font-weight: 600;
+    font-size: 1.2em;
+}
 """
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):