abhilash88

/

age-gender-prediction

like 2

Hi! Thanks for catching that! 🙏

Fixed both issues - just updated the repo with:

• ✅ Complete model.py file
• ✅ Native transformers pipeline support

Super easy usage now:

from transformers import pipeline
classifier = pipeline("age-gender-classification", model="abhilash88/age-gender-prediction")
result = classifier("your_image.jpg")
print(f"Age: {result['age']}, Gender: {result['gender']}")

Try it out and let me know if you hit any issues. 🚀

Hi again! With pipeline("image-classification", model="abhilash88/age-gender-prediction", trust_remote_code=True) i had this issue:

    raise ValueError(
ValueError: Unrecognized configuration class <class 'transformers_modules.abhilash88.age-gender-prediction.fe2695a37455d87bcb01a3171b3e2dcf144966a4.model.AgeGenderConfig'> for this kind of AutoModel: TFAutoModelForImageClassification.
Model type should be one of ConvNextConfig, ConvNextV2Config, CvtConfig, Data2VecVisionConfig, DeiTConfig, EfficientFormerConfig, MobileViTConfig, RegNetConfig, ResNetConfig, SegformerConfig, SwiftFormerConfig, SwinConfig, ViTConfig.

Fixed! Please try to use the following code again

from transformers import pipeline
classifier = pipeline("image-classification", model="abhilash88/age-gender-prediction", trust_remote_code=True)
result = classifier("your_image.jpg")
print(f"Age: {result[0]['age']}, Gender: {result[0]['gender']}")

It works fine for me now.