Spaces:
Running
Running
Update app.py
Browse files
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 |
-
|
|
|
|
|
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="
|
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():
|