diff --git a/server.py b/server.py
index d65998bb..c951da86 100644
--- a/server.py
+++ b/server.py
@@ -855,6 +855,7 @@ else:
     default_text = settings['prompt']
 description = f"\n\n# Text generation lab\nGenerate text using Large Language Models.\n"
 css = ".my-4 {margin-top: 0} .py-6 {padding-top: 2.5rem} #refresh-button {flex: none; margin: 0; padding: 0; min-width: 50px; border: none; box-shadow: none; border-radius: 0} #download-label, #upload-label {min-height: 0}"
+chat_css = ".h-\[40vh\] {height: 66.67vh} .gradio-container {max-width: 800px; margin-left: auto; margin-right: auto} .w-screen {width: unset}"
 suffix = '_pygmalion' if 'pygmalion' in model_name.lower() else ''
 buttons = {}
 gen_events = []
@@ -866,7 +867,7 @@ if args.chat or args.cai_chat:
     if Path(f'logs/persistent.json').exists():
         load_history(open(Path(f'logs/persistent.json'), 'rb').read(), settings[f'name1{suffix}'], settings[f'name2{suffix}'])
 
-    with gr.Blocks(css=css+".h-\[40vh\] {height: 66.67vh} .gradio-container {max-width: 800px; margin-left: auto; margin-right: auto} .w-screen {width: unset}", analytics_enabled=False) as interface:
+    with gr.Blocks(css=css+chat_css, analytics_enabled=False) as interface:
         if args.cai_chat:
             display = gr.HTML(value=generate_chat_html(history['visible'], settings[f'name1{suffix}'], settings[f'name2{suffix}'], character))
         else: