cakiki commited on
Commit
3e900b8
1 Parent(s): 792be34

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,11 +11,11 @@ pd.options.plotting.backend = "plotly"
11
 
12
  TITLE = "Diffusion Professions Cluster Explorer"
13
 
14
- professions = load_from_disk("professions")
15
  professions_df = professions.to_pandas()
16
 
17
  def get_image(model, fname):
18
- return professions.select(professions_df[(professions_df["image_path"]==fname) & (professions_df["model"]==model)].index)["image"][0]
19
 
20
  clusters_dicts = dict(
21
  (num_cl, json.load(open(f"clusters/professions_to_clusters_{num_cl}.json")))
 
11
 
12
  TITLE = "Diffusion Professions Cluster Explorer"
13
 
14
+ professions_dset = load_from_disk("professions")
15
  professions_df = professions.to_pandas()
16
 
17
  def get_image(model, fname):
18
+ return professions_dset.select(professions_df[(professions_df["image_path"]==fname) & (professions_df["model"]==model)].index)["image"][0]
19
 
20
  clusters_dicts = dict(
21
  (num_cl, json.load(open(f"clusters/professions_to_clusters_{num_cl}.json")))