Downloads · 30 days
0
hairstyle-o1
k1rupakaran/hairstyle-o1
hairstyle-o1 is a machine learning model from k1rupakaran. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
--- tags: - fastai - vision - image-classification ---
Access
Public
Updated Mar 8, 2025
Repo size
47 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pkl47 MB · 100%
At a glance
- License
- apache-2.0
- Access
- Public
- Created
- Mar 8, 2025
- Updated
- Mar 8, 2025
- SHA
- babdc367
Model card
From the Hugging Face model README
tags:
- fastai
- vision
- image-classification
My Fine-Tuned Model
This is a fine-tuned ResNet18 model for image classification.
Usage
from fastai.vision.all import *
# Load the model
learn = load_learner('k1rupakaran/hairstyle-o1')
# Make predictions
img = PILImage.create('example.jpg')
pred, idx, probs = learn.predict(img)
print(f"Prediction: {learn.dls.vocab[idx]}, Probability: {probs[idx]:.4f}")
- License
- apache-2.0
- Created
- Mar 8, 2025
- Updated
- Mar 8, 2025