Dominik Hintersdorf commited on
Commit
22110aa
β€’
1 Parent(s): 3ffe17d

fix occurence information only for laion400m models

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -477,7 +477,7 @@ def on_submit_btn_click(model_name, true_name, prompts, images):
477
  len(PROMPTS)
478
  )
479
 
480
- if true_name.lower() in LAION_MEMBERSHIP_OCCURENCE['name'].str.lower().values:
481
  row = LAION_MEMBERSHIP_OCCURENCE[LAION_MEMBERSHIP_OCCURENCE['name'].str.lower() == true_name.lower()]
482
  interpretation = interpretation + OCCURENCE_INFORMATION.format(true_name, row['count'].values[0])
483
 
 
477
  len(PROMPTS)
478
  )
479
 
480
+ if 'laion400m' in model_name.lower() and true_name.lower() in LAION_MEMBERSHIP_OCCURENCE['name'].str.lower().values:
481
  row = LAION_MEMBERSHIP_OCCURENCE[LAION_MEMBERSHIP_OCCURENCE['name'].str.lower() == true_name.lower()]
482
  interpretation = interpretation + OCCURENCE_INFORMATION.format(true_name, row['count'].values[0])
483