Update app.py

app.py

@@ -3,42 +3,51 @@ from utils import *
 from transformers import pipeline
 
 css = """
-button {
-    background-color: #673AB7; /* 设置按钮背景为紫色 */
-    color: white; /* 设置按钮文字颜色为白色 */
-    border-radius: 8px; /* 设置按钮边角为圆角 */
-    padding: 10px 20px; /* 设置按钮内边距 */
-    border: none; /* 移除按钮边框 */
-}
+
 """
 
 ori_model = None
 edit_model = None
 
-with gr.Blocks(css=css) as demo:
-    gr.Markdown("# Model Output Editor")
-    gr.Markdown("This interface takes your input, shows the output of the source model, and then the edited model's output.")
+with gr.Blocks(css=css,theme=gr.themes.Soft(text_size="sm")) as demo:
+    gr.HTML("""<div style="text-align: center; margin: 0 auto;"><p><h1> Knowledge Editing</h1></div>""")
+
+    with gr.Row():
+        gr.Markdown("<p align='center'><a href='https://github.com/zjunlp/EasyEdit'>🔧https://github.com/zjunlp/EasyEdit</a></p>") 
+    
+    with gr.Row():
+        gr.Markdown("### Large Language Models (LLMs) often suffer from knowledge cutoff or fallacy issues, meaning they are unable to recognize unseen events or generate text with incorrect facts due to outdated/noisy data. To address this, many knowledge editing methods for LLMs aim to subtly inject/edit updated knowledge or adjust undesirable behaviors, while minimizing the impact on unrelated inputs.")    
+    
+    
+    with gr.Row():     
+        prompt = gr.Textbox(label="Edit Prompt")     
+        target_new = gr.Textbox(label="Edit Target New")      
+    with gr.Row():
+        button4clear = gr.Button("Clear")
+        button4edit = gr.Button("Edit",variant="primary") 
+    with gr.Row():
+        examples = gr.Examples(
+            examples=[
+                ["Who is the architect for Toodyay Fire Station?","Wong Tung & Sons"],
+                ["Who is Claire Clairmont\'s sister?","Clairmont-Mayer"],
+                ["Which fictional universe is Chlorophyll Kid part of?","Image Universe"]
+            ],
+            examples_per_page=3,
+            inputs=[prompt,target_new],
+        )     
+    with gr.Row():
+        input_text = gr.Textbox(label="Status Information",value="Model editing may take about a minute, please be patient.")
+        
+    with gr.Row():
+        input = gr.Textbox(label="Input Text",lines=3)
     with gr.Row():
         with gr.Column():
-            with gr.Row():     
-                prompt = gr.Textbox(label="Input Prompt",lines=4)
-            with gr.Row():
-                target_new = gr.Textbox(label="Input Target New")
-            with gr.Row():
-                button4clear = gr.Button("Clear")
-                button4edit = gr.Button("Edit")
-            with gr.Row():
-                input_text = gr.Label(label="Status Information",value="The editing process may take up to 30 seconds. Please be patient.")
+            button4gen_ori=gr.Textbox(label="origin output")
         with gr.Column():
-            with gr.Row():
-                input = gr.Textbox(label="Input Text")
-            with gr.Row():
-                button4gen = gr.Button("Generate")
-            with gr.Row():
-                button4gen_ori=gr.Label(label="origin output")
-            with gr.Row():
-                button4gen_edit=gr.Label(label="edited output")
-                
+            button4gen_edit=gr.Textbox(label="edited output")
+    with gr.Row():
+        button4gen = gr.Button("Generate",variant="primary")
+   
     button4clear.click(lambda: ("", ""), outputs=[prompt,target_new])
     button4edit.click(fn=edit, inputs=[prompt,target_new], outputs=input_text)
     button4gen.click(fn=generate, inputs=input, outputs=[button4gen_ori,button4gen_edit])