Update app.py

app.py

@@ -40,7 +40,13 @@ def generate(prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition
         yield output.strip('</s>')
     return output.strip('</s>')
 
-with gr.Blocks() as demo:
+with gr.Blocks(css="""
+    body {background-color: #121212; color: #ffffff;}
+    .gradio-container {background-color: #121212; color: #ffffff;}
+    input, textarea, select, button {background-color: #333333 !important; color: #ffffff !important; border-color: #555555 !important;}
+    .gradio-container button {background-color: #555555 !important;}
+    """) as demo:
+    
     with gr.Row():
         input_text = gr.Textbox(placeholder="Enter your prompt here...", lines=2, max_lines=2, label="Prompt")
         submit_button = gr.Button("Generate")
@@ -48,13 +54,4 @@ with gr.Blocks() as demo:
     
     submit_button.click(fn=generate, inputs=input_text, outputs=output_text)
 
-    demo.style(
-        container_background_color="#121212",
-        text_color="#ffffff",
-        input_background_color="#333333",
-        input_border_color="#555555",
-        button_background_color="#555555",
-        button_text_color="#ffffff",
-    )
-
 demo.launch()