[Update]use custom_css to display

app.py

@@ -16,6 +16,17 @@ is_spaces = True if "SPACE_ID" in os.environ else False
 
 is_shared_ui = False
 
+from css_html_js import custom_css
+
+from about import (
+    CITATION_BUTTON_LABEL,
+    CITATION_BUTTON_TEXT,
+    EVALUATION_QUEUE_TEXT,
+    INTRODUCTION_TEXT,
+    LLM_BENCHMARKS_TEXT,
+    TITLE,
+)
+
 
 def excute_udiff(diffusion_model_id, concept, attacker):
     print(f"my IP is {myip}, my port is {myport}")
@@ -37,12 +48,16 @@ css = '''
 '''
 
 
-with gr.Blocks(css=css) as demo:
-    gr.Markdown("# Demo of UnlearnDiffAtk.")
-    gr.Markdown("### UnlearnDiffAtk is an effective and efficient adversarial prompt generation approach for unlearned diffusion models(DMs).")
-#     gr.Markdown("####For more details, please visit the [project](https://www.optml-group.com/posts/mu_attack), 
-# check the [code](https://github.com/OPTML-Group/Diffusion-MU-Attack), and read the [paper](https://arxiv.org/abs/2310.11868).")
-    gr.Markdown("### Please notice that the process may take a long time, but the results will be saved. You can try it later if it waits for too long.")
+with gr.Blocks(css=custom_css) as demo:
+    gr.HTML(TITLE)
+    gr.Markdown(INTRODUCTION_TEXT, elem_classes="markdown-text")
+
+#     gr.Markdown("# Demo of UnlearnDiffAtk.")
+#     gr.Markdown("### UnlearnDiffAtk is an effective and efficient adversarial prompt generation approach for unlearned diffusion models(DMs).")
+# #     gr.Markdown("####For more details, please visit the [project](https://www.optml-group.com/posts/mu_attack), 
+# # check the [code](https://github.com/OPTML-Group/Diffusion-MU-Attack), and read the [paper](https://arxiv.org/abs/2310.11868).")
+#     gr.Markdown("### Please notice that the process may take a long time, but the results will be saved. You can try it later if it waits for too long.")
+    
 
     with gr.Row() as udiff:
         with gr.Column():