Commit
·
de9e814
1
Parent(s):
1352e18
feat: Add Gradio interface in "Explore PapersWithCode Tasks" tab that displays the plotly plots returned by `task_visualizations`
Browse files
app.py
CHANGED
|
@@ -3,6 +3,7 @@ import pandas as pd
|
|
| 3 |
import logging
|
| 4 |
import re
|
| 5 |
from task_visualizations import TaskVisualizations
|
|
|
|
| 6 |
|
| 7 |
logging.basicConfig(level=logging.INFO)
|
| 8 |
|
|
@@ -109,8 +110,27 @@ with gr.Blocks() as demo:
|
|
| 109 |
with gr.Tab("Explore Repository Representations"):
|
| 110 |
setup_repository_representations_tab(repos, representation_types)
|
| 111 |
with gr.Tab("Explore PapersWithCode Tasks"):
|
| 112 |
-
|
| 113 |
-
|
| 114 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 115 |
|
| 116 |
demo.launch()
|
|
|
|
| 3 |
import logging
|
| 4 |
import re
|
| 5 |
from task_visualizations import TaskVisualizations
|
| 6 |
+
import plotly.graph_objects as go
|
| 7 |
|
| 8 |
logging.basicConfig(level=logging.INFO)
|
| 9 |
|
|
|
|
| 110 |
with gr.Tab("Explore Repository Representations"):
|
| 111 |
setup_repository_representations_tab(repos, representation_types)
|
| 112 |
with gr.Tab("Explore PapersWithCode Tasks"):
|
| 113 |
+
gr.Markdown("## PapersWithCode Tasks Visualization")
|
| 114 |
+
with gr.Row():
|
| 115 |
+
min_task_count = gr.Slider(minimum=1, maximum=100, value=10, step=1, label="Minimum Task Count")
|
| 116 |
+
update_button = gr.Button("Update Plots")
|
| 117 |
+
|
| 118 |
+
with gr.Row():
|
| 119 |
+
with gr.Column():
|
| 120 |
+
gr.Markdown("### All Repositories")
|
| 121 |
+
all_repos_plot = gr.Plot()
|
| 122 |
+
with gr.Column():
|
| 123 |
+
gr.Markdown("### Selected Repositories")
|
| 124 |
+
selected_repos_plot = gr.Plot()
|
| 125 |
+
|
| 126 |
+
def update_plots(min_count):
|
| 127 |
+
all_plot, selected_plot = task_visualizations.display_tasks_sunburst_charts(min_count)
|
| 128 |
+
return all_plot, selected_plot
|
| 129 |
+
|
| 130 |
+
update_button.click(
|
| 131 |
+
fn=update_plots,
|
| 132 |
+
inputs=[min_task_count],
|
| 133 |
+
outputs=[all_repos_plot, selected_repos_plot]
|
| 134 |
+
)
|
| 135 |
|
| 136 |
demo.launch()
|