Update README.md

README.md

@@ -7,22 +7,21 @@ tags:
 # Usage
 
 ```python
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+!pip install KeyBartAdapter
 
-tokenizer = AutoTokenizer.from_pretrained("Adapting/KeyBartAdapter")
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+from models import KeyBartAdapter
 
-model = AutoModelForSeq2SeqLM.from_pretrained("Adapting/KeyBartAdapter")
+model = KeyBartAdapter.from_pretrained('Adapting/KeyBartAdapter',adapter_hid_dim =256)
+tokenizer = AutoTokenizer.from_pretrained("bloomberg/KeyBART")
 ```
-- init model: `9c3ed39c6ed5c7e141363e892d77cf8f589d5999`
-- 21 epoch: `38d17da9d05c05929bd57bd3ddf377fdb436261f`
+- adapter layer hd 256 init model: `9c3ed39c6ed5c7e141363e892d77cf8f589d5999`
 
 **1. inference**
 
-```python
 from transformers import Text2TextGenerationPipeline
 pipe = Text2TextGenerationPipeline(model=model,tokenizer=tokenizer)
 
-
 abstract = '''Non-referential face image quality assessment methods have gained popularity as a pre-filtering step on face recognition systems. In most of them, the quality score is usually designed with face matching in mind. However, a small amount of work has been done on measuring their impact and usefulness on Presentation Attack Detection (PAD). In this paper, we study the effect of quality assessment methods on filtering bona fide and attack samples, their impact on PAD systems, and how the performance of such systems is improved when training on a filtered (by quality) dataset. On a Vision Transformer PAD algorithm, a reduction of 20% of the training dataset by removing lower quality samples allowed us to improve the BPCER by 3% in a cross-dataset test.'''
 
 pipe(abstract)