Update app.py

app.py

@@ -8,6 +8,8 @@ import spaces
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig, TextIteratorStreamer
 
+from ui import css, PLACEHOLDER
+
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
 MODEL_ID = os.environ.get("MODEL_ID")
@@ -101,5 +103,28 @@ gr.ChatInterface(
         gr.Slider(0, 2, 1.1, label="Repetition penalty"),
         gr.Slider(0, 1, 0.95, label="Top P sampling"),
     ],
-    theme=gr.themes.Soft(primary_hue=COLOR),
+    theme=gr.themes.Soft(primary_hue=COLOR).set(
+        background_fill_primary="#020417",
+        background_fill_secondary_dark="#020417",
+        body_background_fill_dark="#020417",
+        block_background_fill_dark="#020417",
+        block_border_width="1px",
+        block_title_background_fill_dark="#15172c",
+        input_background_fill_dark="#15172c",
+        button_secondary_background_fill_dark="#15172c",
+        border_color_accent_dark="#15172c",
+        border_color_primary_dark="#15172c",
+        color_accent_soft_dark="#10132c",
+        code_background_fill_dark="#15172c",
+    ),
+    css=css,
+    retry_btn="Retry",
+    undo_btn="Undo",
+    clear_btn="Clear",
+    submit_btn="Send",
+    chatbot=gr.Chatbot(
+        scale=1,
+        placeholder=PLACEHOLDER,
+        show_copy_button=True
+    )
 ).queue().launch()