loubnabnl HF staff commited on
Commit
f2ed9e8
·
verified ·
1 Parent(s): dc90136

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ Additionally, the model was tasked with finding the topic of each cluster.
18
  @st.cache_data
19
  def load_data(min_score=1, max_score=10):
20
  ds = load_dataset("HuggingFaceTB/FW_clusters_free_topics", split="train", token=HF_TOKEN, num_proc=2)
21
- ds = ds.filter(lambda x: x['educational_score'] <= max_score and x['educational_score'] >= min_score)
22
  return ds
23
 
24
  st.subheader("Cluster information")
 
18
  @st.cache_data
19
  def load_data(min_score=1, max_score=10):
20
  ds = load_dataset("HuggingFaceTB/FW_clusters_free_topics", split="train", token=HF_TOKEN, num_proc=2)
21
+ ds = ds.filter(lambda x: int(x['educational_score']) <= max_score and int(x['educational_score']) >= min_score)
22
  return ds
23
 
24
  st.subheader("Cluster information")