Spaces:
Sleeping
Sleeping
Dominik Hintersdorf
commited on
Commit
·
22110aa
1
Parent(s):
3ffe17d
fix occurence information only for laion400m models
Browse files
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 |
|