Update app.py

app.py

@@ -89,7 +89,9 @@ def genderplot(genlist):
 def age_detector(image):
     pipe = pipeline('image-classification', model="dima806/faces_age_detection", device=0)
     result = pipe(image)
+    print(result)
     filtered_labels = [item['label'] for item in result if item['score'] > 0.85]
+    print(filtered_labels)
     return filtered_labels
 
 def ageplot(agelist):
@@ -150,9 +152,7 @@ def generate_images_plots(prompt, model_name):
         prompt_prefix = "photo of a "
         caption = blip_caption_image(image, prefix=prompt_prefix)
         age = age_detector(image)
-        print(age)
         nsfw = is_nsfw(image)
-        print(nsfw)
         image.save(f"{foldername}/image_{i}.png")
         try:
             skintoneres = stone.process(f"{foldername}/image_{i}.png", return_report_image=False)