File size: 794 Bytes
b1f36e1 321cccd b1f36e1 321cccd 1f45022 b1f36e1 1f45022 321cccd b1f36e1 321cccd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
---
language: en
tags:
- image-classification
- fastai
- vision
datasets:
- cats
license: mit
library_name: fastai
---
# Cat Image Classifier
This model classifies whether an input image contains a cat or not. You can use the model through:
1. The interface above - just upload an image and get predictions
2. Programmatically using the code below
## Programmatic Usage
```python
from fastai.learner import load_learner
from huggingface_hub import hf_hub_download
# Download the model
model_path = hf_hub_download(repo_id="RamyKhorshed/Lesson2FastAi", filename="model.pkl")
model = load_learner(model_path)
# Make a prediction
img_path = "path/to/your/image.jpg"
pred, pred_idx, probs = model.predict(img_path)
print(f"Prediction: {pred}")
print(f"Confidence: {probs[pred_idx]:.4f}")
```
|