File size: 629 Bytes
1776fe8 4cc0757 4f98676 44d08ce 4f98676 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
# MedCPT
###### LingYi system pre training medical model
###### Prease load the model from [**CPT**](https://huggingface.co/fnlp/cpt-large)
## Usage
```python
>>> from modeling_cpt import CPTForConditionalGeneration
>>> from transformers import BertTokenizer
>>> tokenizer = BertTokenizer.from_pretrained("WENGSYX/MedCPT")
>>> model = CPTForConditionalGeneration.from_pretrained("WENGSYX/MedCPT")
>>> inputs = tokenizer.encode("医生你好，腹泻难受应该怎么办？", return_tensors='pt')
>>> pred_ids = model.generate(input_ids, num_beams=4, max_length=20)
>>> print(tokenizer.convert_ids_to_tokens(pred_ids[i]))
``` |