Spaces:
Running
Running
Nikolay Yankovich commited on
Commit ·
00e8206
1
Parent(s): 23e7f59
fix
Browse files
app.py
CHANGED
|
@@ -5,14 +5,13 @@ from sklearn.ensemble import RandomForestRegressor
|
|
| 5 |
from sklearn.metrics import r2_score
|
| 6 |
import matplotlib.pyplot as plt
|
| 7 |
|
| 8 |
-
# Загружаем датасет один раз
|
| 9 |
ds = load_dataset("QSBench/QSBench-Core-v1.0.0-demo")
|
| 10 |
|
| 11 |
-
# Функция для отображения таблицы
|
| 12 |
def show_data(split):
|
| 13 |
try:
|
| 14 |
df = pd.DataFrame(ds[split])
|
| 15 |
-
# Показываем первые 10 строк, все колонки
|
| 16 |
return df.head(10)
|
| 17 |
except Exception as e:
|
| 18 |
return f"Error loading data: {e}"
|
|
@@ -58,16 +57,9 @@ with gr.Blocks(title="QSBench Demo Explorer") as demo:
|
|
| 58 |
label="Choose a split",
|
| 59 |
value="train"
|
| 60 |
)
|
| 61 |
-
data_table = gr.Dataframe(
|
| 62 |
-
|
| 63 |
-
|
| 64 |
-
)
|
| 65 |
-
split_selector.change(
|
| 66 |
-
fn=show_data,
|
| 67 |
-
inputs=split_selector,
|
| 68 |
-
outputs=data_table
|
| 69 |
-
)
|
| 70 |
-
# Загружаем данные по умолчанию при старте
|
| 71 |
demo.load(fn=lambda: show_data("train"), outputs=data_table)
|
| 72 |
|
| 73 |
with gr.TabItem("Model Demo"):
|
|
|
|
| 5 |
from sklearn.metrics import r2_score
|
| 6 |
import matplotlib.pyplot as plt
|
| 7 |
|
| 8 |
+
# Загружаем датасет один раз
|
| 9 |
ds = load_dataset("QSBench/QSBench-Core-v1.0.0-demo")
|
| 10 |
|
| 11 |
+
# Функция для отображения таблицы
|
| 12 |
def show_data(split):
|
| 13 |
try:
|
| 14 |
df = pd.DataFrame(ds[split])
|
|
|
|
| 15 |
return df.head(10)
|
| 16 |
except Exception as e:
|
| 17 |
return f"Error loading data: {e}"
|
|
|
|
| 57 |
label="Choose a split",
|
| 58 |
value="train"
|
| 59 |
)
|
| 60 |
+
data_table = gr.Dataframe(label="First 10 rows", interactive=False)
|
| 61 |
+
split_selector.change(fn=show_data, inputs=split_selector, outputs=data_table)
|
| 62 |
+
# Загружаем данные по умолчанию
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 63 |
demo.load(fn=lambda: show_data("train"), outputs=data_table)
|
| 64 |
|
| 65 |
with gr.TabItem("Model Demo"):
|