Update app.py

app.py

@@ -9,13 +9,6 @@ from threading import Thread
 HF_TOKEN = os.environ.get("HF_TOKEN", None)
 
 
-DESCRIPTION = '''
-<div>
-<h1 style="text-align: center;">GPT 4o</h1>
-</div>
-'''
-
-
 # Load the tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.3")
 model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.3", device_map="auto")
@@ -63,11 +56,11 @@ def gpt4o_mistral(message: str,
         
 
 # Gradio block
-chatbot=gr.Chatbot(height=450, label='Gradio ChatInterface')
+chatbot=gr.Chatbot(height=450, label='Gradio ChatInterface', theme="xiaobaiyuan/theme_brief")
 
 with gr.Blocks(fill_height=True) as demo:
     
-    gr.Markdown(DESCRIPTION)
+
     gr.ChatInterface(
         fn=gpt4o_mistral,
         chatbot=chatbot,
@@ -95,7 +88,7 @@ with gr.Blocks(fill_height=True) as demo:
             ['Justify why a penguin might make a good king of the jungle.']
             ],
         cache_examples=False,
-        theme="xiaobaiyuan/theme_brief"
+       
                      )