evijit HF staff commited on
Commit
1b7f205
·
verified ·
1 Parent(s): 528bb1f

Update app.py

Browse files

Fix plot function

Files changed (1) hide show
  1. app.py +2 -2
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(age)
236
- nsfws.append(nsfw)
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: