Update app.py

app.py

@@ -9,14 +9,7 @@ import os, json
 from sys import argv
 from vllm import LLM, SamplingParams
 import vllm
-
-from huggingface_hub import login
-TOKEN = os.environ.get("TOKEN", None)
-login(token=TOKEN)
-
-print("transformers version:", transformers.__version__)
-print("vllm version:", vllm.__version__)
-print("gradio version:", gr.__version__)
+import re
 
 
 def load_model_processor(model_path):
@@ -31,7 +24,7 @@ def load_model_processor(model_path):
 model_path1 = "SeaLLMs/SeaLLMs-Audio-7B"
 model1, processor1 = load_model_processor(model_path1)
 
-def response_to_audio(audio_url, text, model=None, processor=None, temperature = 0.1,repetition_penalty=1.1, top_p = 0.9,max_new_tokens = 2048):
+def response_to_audio(audio_url, text, model=None, processor=None, temperature = 0,repetition_penalty=1.1, top_p = 0.9,max_new_tokens = 2048):
     if text == None:
         conversation = [
             {"role": "user", "content": [
@@ -80,29 +73,55 @@ def response_to_audio(audio_url, text, model=None, processor=None, temperature =
 def clear_inputs():
     return None, "", ""
 
+def contains_chinese(text):
+    # Regular expression for Chinese characters
+    chinese_char_pattern = re.compile(r'[\u4e00-\u9fff]')
+    return bool(chinese_char_pattern.search(text))
+
 def compare_responses(audio_url, text):
+    if contains_chinese(text): 
+        return "Caution! This demo does not support Chinese!"
+    
     response1 = response_to_audio(audio_url, text, model1, processor1)
+    if contains_chinese(response1): 
+        return "ERROR! Try another example!"
+
     return response1
 
 with gr.Blocks() as demo:
     # gr.Markdown(f"Evaluate {model_path1}")
-    # gr.Markdown("""<p align="center"><img src="images/seal_logo.png" style="height: 80px"/><p>""")
+    gr.HTML("""<p align="center"><img src="https://DAMO-NLP-SG.github.io/SeaLLMs-Audio/static/images/seallm-audio-logo.png" style="height: 80px"/><p>""")
     # gr.Image("images/seal_logo.png", elem_id="seal_logo", show_label=False,height=80,show_fullscreen_button=False)
-    # gr.Markdown("""<center><font size=8>SeaLLMs-Audio Demo</center>""")
-    gr.Markdown("""# SeaLLMs-Audio Demo""")
-    gr.Markdown(
-        """\
-<center><font size=4>This WebUI is based on SeaLLMs-Audio-7B-Chat, developed by Alibaba DAMO Academy.<br>
-    You can interact with the chatbot in <b>English, Chinese, Indonesian, Thai, or Vietnamese</b>.<br>
-    For the input, you can input <b>audio and/or text</center>.""")
-
-    # Links with proper formatting
-    gr.Markdown(
-        """<center><font size=4>
-        <a href="https://huggingface.co/SeaLLMs/SeaLLMs-v3-7B-Chat">[Website]</a> &nbsp; 
-        <a href="https://huggingface.co/SeaLLMs/SeaLLMs-v3-7B-Chat">[Model🤗]</a> &nbsp; 
-        <a href="https://github.com/liuchaoqun/SeaLLMs-Audio">[Github]</a>
-        </center>""",
+    gr.HTML("""<h1 align="center" id="space-title">SeaLLMs-Audio-Demo</h1>""")
+#     gr.Markdown(
+#         """\
+# <center><font size=4>This WebUI is based on SeaLLMs-Audio-7B, developed by Alibaba DAMO Academy.<br>
+#     You can interact with the chatbot in <b>English, Chinese, Indonesian, Thai, or Vietnamese</b>.<br>
+#     For the input, you can input <b>audio and/or text</center>.""")
+
+#     # Links with proper formatting
+#     gr.Markdown(
+#         """<center><font size=4>
+#         <a href="https://huggingface.co/SeaLLMs/SeaLLMs-v3-7B-Chat">[Website]</a> &nbsp; 
+#         <a href="https://huggingface.co/SeaLLMs/SeaLLMs-Audio-7B">[Model🤗]</a> &nbsp; 
+#         <a href="https://github.com/DAMO-NLP-SG/SeaLLMs-Audio">[Github]</a>
+#         </center>""",
+#     )
+
+    gr.HTML(
+        """<div style="text-align: center; font-size: 16px;">
+        This WebUI is based on <a href="https://huggingface.co/SeaLLMs/SeaLLMs-Audio-7B">SeaLLMs-Audio-7B</a>, developed by Alibaba DAMO Academy.<br>
+        You can interact with the chatbot in <b>English, Indonesian, Thai, or Vietnamese</b>.<br>
+        For the input, you can provide <b>audio and/or text</b>.
+        </div>"""
+    )
+    
+    gr.HTML(
+        """<div style="text-align: center; font-size: 16px;">
+        <a href="https://DAMO-NLP-SG.github.io/SeaLLMs-Audio/">[Website]</a> &nbsp; 
+        <a href="https://huggingface.co/SeaLLMs/SeaLLMs-Audio-7B">[Model🤗]</a> &nbsp; 
+        <a href="https://github.com/DAMO-NLP-SG/SeaLLMs-Audio">[Github]</a>
+        </div>"""
     )
 
     # gr.Markdown(insturctions)