|
|
--- |
|
|
language: en |
|
|
license: mit |
|
|
tags: |
|
|
- skin-care |
|
|
- image-classification |
|
|
- cnn |
|
|
- deep-learning |
|
|
datasets: |
|
|
- custom |
|
|
pipeline_tag: image-segmentation |
|
|
--- |
|
|
|
|
|
# 🧴 Skin-Care Classification Model |
|
|
|
|
|
This model was trained to classify different skin conditions using deep learning. |
|
|
|
|
|
## 🔎 Model Details |
|
|
- **Framework**: TensorFlow / PyTorch (choose one) |
|
|
- **Input**: Skin images |
|
|
- **Output**: Predicted class (e.g., acne, normal, eczema, etc.) |
|
|
- **Author**: [BARATHVISWARAJ](https://huggingface.co/BARATHVISWARAJ) |
|
|
|
|
|
## 📊 Training |
|
|
- Dataset: Custom skin-care dataset |
|
|
- Image size: 224x224 |
|
|
- Optimizer: Adam |
|
|
- Loss: CrossEntropy |
|
|
- Epochs: (fill in your training details) |
|
|
|
|
|
## 💡 Usage |
|
|
```python |
|
|
from transformers import AutoModel, AutoTokenizer |
|
|
from PIL import Image |
|
|
import torch |
|
|
|
|
|
model = AutoModel.from_pretrained("BARATHVISWARAJ/Skin-Care") |
|
|
|
|
|
# Example: loading image and running prediction |
|
|
# (replace with actual preprocessing code) |
