Scott K commited on
Commit
2753a77
1 Parent(s): c9ec638

updated model card

Browse files
Files changed (1) hide show
  1. README.md +27 -7
README.md CHANGED
@@ -1,12 +1,12 @@
1
  ---
2
  tags:
3
- - fastai
4
- - resnet
5
- - computer-vision
6
- - classification
7
- - binary-classification
8
  license:
9
- - cc0-1.0
10
  ---
11
 
12
  # Resnet34 Pokemon Card Classifier
@@ -15,4 +15,24 @@ license:
15
 
16
  This is a resnet34 model fine-tuned with fastai to [classify real and fake Pokemon cards (dataset)](https://www.kaggle.com/datasets/ongshujian/real-and-fake-pokemon-cards).
17
 
18
- ## Intended uses & limitation
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  tags:
3
+ - fastai
4
+ - resnet
5
+ - computer-vision
6
+ - classification
7
+ - binary-classification
8
  license:
9
+ - cc0-1.0
10
  ---
11
 
12
  # Resnet34 Pokemon Card Classifier
 
15
 
16
  This is a resnet34 model fine-tuned with fastai to [classify real and fake Pokemon cards (dataset)](https://www.kaggle.com/datasets/ongshujian/real-and-fake-pokemon-cards).
17
 
18
+ Here is a colab notebook that shows how the model was trained and pushed to the hub: [link](https://github.com/mindwrapped/pokemon-card-checker/blob/main/pokemon_card_checker.ipynb).
19
+
20
+ ## Intended uses & limitation
21
+
22
+ This model is trained to identify real vs fake cards based on the backs of the cards, not the front.
23
+
24
+ ## How to use
25
+
26
+ ```python
27
+ from huggingface_hub import from_pretrained_fastai
28
+
29
+ # Pull model from hub
30
+ learn = from_pretrained_fastai('hugginglearners/pokemon-card-checker')
31
+
32
+ # Get prediction for this image
33
+ pred_label, _, scores = learn.predict(img)
34
+ ```
35
+
36
+ ## Training data
37
+
38
+ Dataset located here: [link](https://www.kaggle.com/datasets/ongshujian/real-and-fake-pokemon-cards).