| 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}") | |
| ``` | |