Hayloo9838 commited on
Commit
e283e1c
·
verified ·
1 Parent(s): 07ff5a5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
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. The current model.py uses your webcam to track what is going on, but you can change it to do whatever.
 
 
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