hi-sammy commited on
Commit
965425f
·
verified ·
1 Parent(s): 6a60c24

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -7
app.py CHANGED
@@ -10,17 +10,19 @@ def filter_records(records, category):
10
  filtered = filtered.drop("Selective Alignment", axis=1)
11
  return filtered.iloc[:, 1:]
12
 
13
- with gr.Blocks() as demo:
 
 
14
  with gr.Row():
15
- button_celebrity = gr.Button(value="Celebrity")
16
- button_style = gr.Button(value="Style")
17
- button_ip = gr.Button(value="IP")
18
- button_nsfw = gr.Button(value="NSFW")
19
- button_all = gr.Button(value="Overall")
20
 
21
  # Initialize dataframe with Celebrity data
22
  celebrity_data = filter_records(df, "Celebrity")
23
- dataframe = gr.DataFrame(celebrity_data, column_widths="9%")
24
 
25
  # Create separate functions for each category that always start with the original data
26
  def filter_celebrity():
 
10
  filtered = filtered.drop("Selective Alignment", axis=1)
11
  return filtered.iloc[:, 1:]
12
 
13
+ theme = gr.themes.Base()
14
+
15
+ with gr.Blocks(theme=theme) as demo:
16
  with gr.Row():
17
+ button_celebrity = gr.Button(value="Celebrity", variant="primary")
18
+ button_style = gr.Button(value="Style", variant="primary")
19
+ button_ip = gr.Button(value="IP", variant="primary")
20
+ button_nsfw = gr.Button(value="NSFW", variant="primary")
21
+ button_all = gr.Button(value="Overall", variant="primary")
22
 
23
  # Initialize dataframe with Celebrity data
24
  celebrity_data = filter_records(df, "Celebrity")
25
+ dataframe = gr.DataFrame(celebrity_data, column_widths="8%", elem_classes="light-mode")
26
 
27
  # Create separate functions for each category that always start with the original data
28
  def filter_celebrity():