add router hint and highlight fast demo

app.py

@@ -6,7 +6,7 @@
 #    Email         : zhangfan@baai.ac.cn
 #    Institute     : Beijing Academy of Artificial Intelligence (BAAI)
 #    Create On     : 2023-12-11 15:34
-#    Last Modified : 2023-12-29 08:04
+#    Last Modified : 2024-01-04 10:15
 #    File Name     : app.py
 #    Description   :
 #
@@ -36,14 +36,19 @@ parser.add_argument("--disable-generate", action="store_true")
 args = parser.parse_args()
 
 
+# <h2 align='center'>[<a href='https://emu.ssi.plus' target='_blank' rel='noopener'>FAST demo</a>]</h2>
 if __name__ == "__main__":
     title = "Emu2: Generative Multimodal Models are In-Context Learners<br> \
             <h2 align='center'> \
             [<a href='https://baaivision.github.io/emu2' target='_blank' rel='noopener'>project page</a>] \
             [<a href='https://github.com/baaivision/Emu' target='_blank' rel='noopener'>code</a>] \
             [<a href='https://arxiv.org/abs/2312.13286' target='_blank' rel='noopener'>paper</a>] \
-            [<a href='https://emu.ssi.plus' target='_blank' rel='noopener'>FAST demo</a>] \
             </h2> \
+            <div align='center'> \
+            <font size=4>Experience Emu2 with 👉</font> \
+            <font size=6>[<b><a href='https://emu.ssi.plus' target='_blank' rel='noopener'>FAST demo</a>]</b></font> \
+            <font size=4>👈 <b>WAY MORE FASTER!!!</b></font> \
+            </div> \
             <h3 align='center'> \
             🤗HF models: \
             <a href='https://huggingface.co/BAAI/Emu2' target='_blank' rel='noopener'>Emu2</a> | \

demo/chat_frontend.py

@@ -6,7 +6,7 @@
 #    Email         : zhangfan@baai.ac.cn
 #    Institute     : Beijing Academy of Artificial Intelligence (BAAI)
 #    Create On     : 2023-12-12 18:05
-#    Last Modified : 2023-12-22 10:29
+#    Last Modified : 2024-01-04 09:42
 #    File Name     : chat_frontend.py
 #    Description   :
 #
@@ -23,7 +23,7 @@ import gradio as gr
 from .meta import ConvMeta, Role, DataMeta
 from .utils import extract_frames
 from .utils import frontend_logger as logging
-from .constants import TERM_OF_USE, CHAT_GUIDANCE, RECOMMEND
+from .constants import CHAT_ROUTER, TERM_OF_USE, CHAT_GUIDANCE, RECOMMEND
 
 CONTROLLER_URL = ""
 
@@ -164,6 +164,7 @@ def build_chat(args):
     with gr.Blocks(title="Emu", theme=gr.themes.Default(primary_hue="blue", secondary_hue="blue")) as demo:
         state = gr.State()
 
+        gr.Markdown(CHAT_ROUTER)
         gr.Markdown(CHAT_GUIDANCE)
         gr.Markdown(RECOMMEND)
 

demo/constants.py

@@ -54,3 +54,6 @@ LICENSE = "### License\n📌The service is a research preview intended for non-c
 CHAT_GUIDANCE = "<font size=4>💡<b><u>NOTE</u></b>💡: You can <b><u>_ONLY_</u></b> input one modality at a time. Please <b><u>_CLICK_</u></b> Add button after entering one modality input. And then input the <b><u>_NEXT_</u></b> modality.</font>"
 GEN_GUIDANCE = "<font size=4>💡<b><u>NOTE</u></b>💡: You can <b><u>_ONLY_</u></b> input one modality at a time (except for <b><u>_location-binded_</u></b> inputs). Please <b><u>_CLICK_</u></b> Add button after entering one modality input. And then input the <b><u>_NEXT_</u></b> modality.</font>"
 RECOMMEND = "<font size=2>Strongly <b><u>_RECOMMAND_</u></b> reading the [user guide](https://jwolpxeehx.feishu.cn/docx/RYHNd1tvEo8k8Mx9HeMcvvxWnvZ) before trying out this demo.</font>"
+
+CHAT_ROUTER = "<font size=4>🚏<b><u>ROUTER</u></b>🚏:This tab can <b><u>_ONLY_</u></b> generate <b><u>_TEXT_</u></b> output. If you want to generate images, please use <b>_Multimodal Generation_</b>.</font>"
+GEN_ROUTER = "<font size=4>🚏<b><u>ROUTER</u></b>🚏:This tab can <b><u>_ONLY_</u></b> generate <b><u>_IMAGE_</u></b> output. If you want to generate texts, please use <b>_Multimodal Chat_</b>.</font>"

demo/generation_frontend.py

@@ -6,7 +6,7 @@
 #    Email         : zhangfan@baai.ac.cn
 #    Institute     : Beijing Academy of Artificial Intelligence (BAAI)
 #    Create On     : 2023-12-11 15:35
-#    Last Modified : 2023-12-22 10:29
+#    Last Modified : 2024-01-04 09:43
 #    File Name     : generation_frontend.py
 #    Description   :
 #
@@ -21,7 +21,7 @@ import requests
 
 import gradio as gr
 
-from .constants import EVA_IMAGE_SIZE
+from .constants import EVA_IMAGE_SIZE, GEN_ROUTER
 from .meta import ConvMeta, Role, DataMeta
 from .utils import frontend_logger as logging
 from .constants import TERM_OF_USE, GEN_GUIDANCE, RECOMMEND
@@ -161,6 +161,7 @@ def build_generation(args):
         state = gr.State()
 
         gr.Markdown("<font size=5><center><b>This demo</b> can accept a mix of <b><u>_texts_</u></b>, <b><u>_locations_</u></b> and <b><u>_images_</u></b> as input, and generating images in context</center></font>")
+        gr.Markdown(GEN_ROUTER)
         gr.Markdown(GEN_GUIDANCE)
         gr.Markdown(RECOMMEND)
         gr.Markdown("<font size=4>💡<b><u>Tips</b></u>💡:</font> To achieve better generation quality\n \