msiron commited on
Commit
fa131a7
·
1 Parent(s): 6f4e929

memory fix double variables, fix var name

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -30,6 +30,7 @@ dataset = load_dataset(
30
 
31
  # Convert the train split to a pandas DataFrame
32
  train_df = dataset.to_pandas()
 
33
 
34
 
35
  def create_phase_diagram(
@@ -42,7 +43,7 @@ def create_phase_diagram(
42
  entries = [
43
  ComputedStructureEntry(
44
  Structure(
45
- [x.tolist() for x in df.iloc[0]["lattice_vectors"].tolist()],
46
  row["species_at_sites"],
47
  row["cartesian_site_positions"],
48
  coords_are_cartesian=True,
 
30
 
31
  # Convert the train split to a pandas DataFrame
32
  train_df = dataset.to_pandas()
33
+ del dataset
34
 
35
 
36
  def create_phase_diagram(
 
43
  entries = [
44
  ComputedStructureEntry(
45
  Structure(
46
+ [x.tolist() for x in train_df.iloc[0]["lattice_vectors"].tolist()],
47
  row["species_at_sites"],
48
  row["cartesian_site_positions"],
49
  coords_are_cartesian=True,