Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse filesFix plot function
app.py
CHANGED
|
@@ -232,8 +232,8 @@ def generate_images_plots(prompt, model_name):
|
|
| 232 |
except:
|
| 233 |
skintones.append(None)
|
| 234 |
genders.append(genderfromcaption(caption))
|
| 235 |
-
ages.append(
|
| 236 |
-
nsfws.append(
|
| 237 |
return images, skintoneplot(skintones), genderplot(genders), ageplot(ages), nsfwplot(nsfws)
|
| 238 |
|
| 239 |
with gr.Blocks(title="Demographic bias in Text-to-Image Generation Models") as demo:
|
|
|
|
| 232 |
except:
|
| 233 |
skintones.append(None)
|
| 234 |
genders.append(genderfromcaption(caption))
|
| 235 |
+
ages.append(age_detector(image)) # Call age_detector function
|
| 236 |
+
nsfws.append(is_nsfw(image)) # Call is_nsfw function
|
| 237 |
return images, skintoneplot(skintones), genderplot(genders), ageplot(ages), nsfwplot(nsfws)
|
| 238 |
|
| 239 |
with gr.Blocks(title="Demographic bias in Text-to-Image Generation Models") as demo:
|