Update app.py

app.py

@@ -10,17 +10,19 @@ def filter_records(records, category):
         filtered = filtered.drop("Selective Alignment", axis=1)
     return filtered.iloc[:, 1:]
 
-with gr.Blocks() as demo:
+theme = gr.themes.Base()
+
+with gr.Blocks(theme=theme) as demo:
     with gr.Row():
-        button_celebrity = gr.Button(value="Celebrity")
-        button_style = gr.Button(value="Style")
-        button_ip = gr.Button(value="IP")
-        button_nsfw = gr.Button(value="NSFW")
-        button_all = gr.Button(value="Overall")
+        button_celebrity = gr.Button(value="Celebrity", variant="primary")
+        button_style = gr.Button(value="Style", variant="primary")
+        button_ip = gr.Button(value="IP", variant="primary")
+        button_nsfw = gr.Button(value="NSFW", variant="primary")
+        button_all = gr.Button(value="Overall", variant="primary")
     
     # Initialize dataframe with Celebrity data
     celebrity_data = filter_records(df, "Celebrity")
-    dataframe = gr.DataFrame(celebrity_data, column_widths="9%")
+    dataframe = gr.DataFrame(celebrity_data, column_widths="8%", elem_classes="light-mode")
     
     # Create separate functions for each category that always start with the original data
     def filter_celebrity():