RamyKhorshed
commited on
Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
@@ -12,14 +12,11 @@ library_name: fastai
|
|
12 |
|
13 |
# Cat Image Classifier
|
14 |
|
15 |
-
|
16 |
-
|
|
|
17 |
|
18 |
-
##
|
19 |
-
- Detecting cats in photographs
|
20 |
-
- Educational purposes for learning FastAI and image classification
|
21 |
-
|
22 |
-
## How to use
|
23 |
|
24 |
```python
|
25 |
from fastai.learner import load_learner
|
@@ -30,17 +27,8 @@ model_path = hf_hub_download(repo_id="RamyKhorshed/Lesson2FastAi", filename="mod
|
|
30 |
model = load_learner(model_path)
|
31 |
|
32 |
# Make a prediction
|
33 |
-
# Replace with your image path
|
34 |
img_path = "path/to/your/image.jpg"
|
35 |
pred, pred_idx, probs = model.predict(img_path)
|
36 |
print(f"Prediction: {pred}")
|
37 |
print(f"Confidence: {probs[pred_idx]:.4f}")
|
38 |
```
|
39 |
-
|
40 |
-
## Limitations
|
41 |
-
- Works best with clear, front-facing photos of cats
|
42 |
-
- May not perform well with unusual angles or partially visible cats
|
43 |
-
- Designed for general cat detection, not breed identification
|
44 |
-
|
45 |
-
## Training
|
46 |
-
This model was trained using FastAI on a dataset of cat and non-cat images.
|
|
|
12 |
|
13 |
# Cat Image Classifier
|
14 |
|
15 |
+
This model classifies whether an input image contains a cat or not. You can use the model through:
|
16 |
+
1. The interface above - just upload an image and get predictions
|
17 |
+
2. Programmatically using the code below
|
18 |
|
19 |
+
## Programmatic Usage
|
|
|
|
|
|
|
|
|
20 |
|
21 |
```python
|
22 |
from fastai.learner import load_learner
|
|
|
27 |
model = load_learner(model_path)
|
28 |
|
29 |
# Make a prediction
|
|
|
30 |
img_path = "path/to/your/image.jpg"
|
31 |
pred, pred_idx, probs = model.predict(img_path)
|
32 |
print(f"Prediction: {pred}")
|
33 |
print(f"Confidence: {probs[pred_idx]:.4f}")
|
34 |
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|