Update app.py

app.py

@@ -387,12 +387,16 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css, title="PEEB") as demo:
     gt_class = gr.State("")
     
     with gr.Column():
-        title_text = gr.Markdown("# PEEB - demo")
+        title_text = gr.Markdown("# A classifier with Part-based Explainable and Editable Bottleneck (PEEB) | Demo")
         gr.Markdown(
             """
-            - In this demo a demo for PEEB paper (NAACL finding 2024). 
-            - paper: https://arxiv.org/abs/2403.05297
-            - code: https://github.com/anguyen8/peeb/tree/inspect_ddp
+            Here is a PEEB classifier pre-trained on Bird-11K and finetuned on CUB-200 (see our [NAACL 2024 paper](https://arxiv.org/abs/2403.05297) and [code](https://github.com/anguyen8/peeb/tree/inspect_ddp)).\n The demo shows how one runs PEEB on an existing image and edit descriptors to modify the classifier (without any re-training).
+
+            Steps:
+            1. **Select an image**. Then, PEEB will show its grounded explanations and the top-1 predicted label with associated softmax confidence score.
+            2. **Hover mouse over text descriptors** to see the corresponding region used to match to each text descriptor.
+            3. **Edit the text under [Custom Descriptions]()** which correspond to one extra, new class (i.e. 200+1 = 201). Further editing will overwrite this class' descriptors.
+            4. **Click on Predict** to see the grounded explanations and the top-1 label for the newly modified CUB-201 classifier.            
             """
         )
 
@@ -401,7 +405,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css, title="PEEB") as demo:
         
         gr.Markdown("## Select an image to start!")
         image_gallery = gr.Gallery(value=IMAGE_GALLERY, label=None, preview=False, allow_preview=False, columns=10, height=250)
-        gr.Markdown("### Custom descritions: \n The first row should be **class name: {some name};**, where you can name your descriptions. \n For the remianing descriptions, please use **;** to separate the descriptions for each part, and use the format **{part name}: {descriptions}**. \n Note that you can delete a part completely, in such cases, all descriptions will remove the corresponding part.")
+        gr.Markdown("### Custom descriptions: \n The first row should be **class name: {some name};**, the name of your 201th class. \n For the 12 part descriptors, please use **;** to separate the descriptions for each part, and use the format **{part name}: {descriptions}**. \n Note: you can delete a row for some part (e.g. *nape*) completely and that part will be removed from all 201 classes in the classifier.")
         
         with gr.Row():
             with gr.Column():