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:
|