gabrielsantosrv commited on
Commit
726ff2b
1 Parent(s): 4ed9c94

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -4
README.md CHANGED
@@ -21,11 +21,10 @@ This model holds the state of the art in many zero-shot tasks involving images
21
 
22
  ## How to use
23
  ```python
24
- from models.open_clip_wrapper import OpenCLIPWrapper
25
- from utils.capivara_utils import download_pretrained_from_hf
26
 
27
- model_path = download_pretrained_from_hf(model_id="hiaac-nlp/CAPIVARA")
28
- model = OpenCLIPWrapper.load_from_checkpoint(model_path, strict=False).model
29
  ```
30
 
31
  For more details refer to [Github repo](https://github.com/hiaac-nlp/CAPIVARA/).
 
21
 
22
  ## How to use
23
  ```python
24
+ import open_clip
 
25
 
26
+ model, preprocess_train, preprocess_val = open_clip.create_model_and_transforms('hf-hub:hiaac-nlp/CAPIVARA')
27
+ tokenizer = open_clip.get_tokenizer('hf-hub:hiaac-nlp/CAPIVARA')
28
  ```
29
 
30
  For more details refer to [Github repo](https://github.com/hiaac-nlp/CAPIVARA/).