Spaces:
Runtime error
Runtime error
add hf token from env variable
Browse files
app.py
CHANGED
@@ -9,8 +9,10 @@ from pymatgen.core.composition import Composition
|
|
9 |
from pymatgen.entries.computed_entries import (ComputedStructureEntry,
|
10 |
GibbsComputedStructureEntry)
|
11 |
|
|
|
|
|
12 |
# Load only the train split of the dataset
|
13 |
-
dataset = load_dataset("LeMaterial/leDataset", split="train")
|
14 |
|
15 |
# Convert the train split to a pandas DataFrame
|
16 |
train_df = dataset.to_pandas()
|
|
|
9 |
from pymatgen.entries.computed_entries import (ComputedStructureEntry,
|
10 |
GibbsComputedStructureEntry)
|
11 |
|
12 |
+
HF_TOKEN = os.environ.get("HF_TOKEN")
|
13 |
+
|
14 |
# Load only the train split of the dataset
|
15 |
+
dataset = load_dataset("LeMaterial/leDataset", token=HF_TOKEN, split="train")
|
16 |
|
17 |
# Convert the train split to a pandas DataFrame
|
18 |
train_df = dataset.to_pandas()
|