This directory contains the trained Machine Learning models used in the Healing Bloom backend system. These models power core functionalities such as skin disease detection, product recommendation, and acne condition analysis.
Models implemented for this task:
- π§ ResNet50
- β‘ EfficientNet
- π§ͺ Custom CNN
- π± DenseNet
These models classify user-uploaded images into various skin disease categories with high accuracy and robustness.
Utilizes the outputs of the disease identification models and additional skin features (e.g., type, condition, sensitivity) to recommend personalized skincare products.
Model used:
- π EfficientNet
This model detects acne levels, including type and spread, using face images. Outputs are then mapped to product suggestions and care routines.
Due to GitHub's 100MB file limit, the following model files are excluded from this repository:
final_model.keras- Other
.h5or.kerasfiles exceeding size limits
You can obtain the model files via:
- Google Drive / S3 (Link here when available)
- Git LFS (if re-enabled)
These models are loaded into the backend using TensorFlow or Keras and are used in FastAPI endpoints for:
- Disease prediction
- Feature-based product recommendation
- Face color and acne evaluation
Ensure you have the following installed:
pip install tensorflow keras opencv-python pillow