Update README.md
Browse files
README.md
CHANGED
@@ -67,7 +67,9 @@ To use this model you will need the pytorch and transformers libraries:
|
|
67 |
```bash
|
68 |
pip install torch transformers torchvision
|
69 |
```
|
70 |
-
Then go to the files of this HF page and simply use model.py in your python scripts, and that is it.
|
|
|
|
|
71 |
|
72 |
## Limitations
|
73 |
|
|
|
67 |
```bash
|
68 |
pip install torch transformers torchvision
|
69 |
```
|
70 |
+
Then go to the files of this HF page and simply use model.py in your python scripts, and that is it.
|
71 |
+
The current model.py will get an image from the internet, perform the processing, then make a heatmap of the attention (what the model cares about) and return its guess and confidence.
|
72 |
+
On that animated image, it gets a pretty underestimating confidence of 30%, but that is because the dataset is insanely limited and i had to resort to augmentation, which is a game changer.
|
73 |
|
74 |
## Limitations
|
75 |
|