Update README.md

README.md

@@ -5,12 +5,13 @@ license: apache-2.0
 
 tags:
 - bart
+- biobart
 - biomedical
 
 inference: true
 
 widget:
-- text: ""
+- text: "Influenza is a <mask> disease."
 
 ---
 # Yuyuan-Bart-139M, one model of [Fengshenbang-LM](https://github.com/IDEA-CCNL/Fengshenbang-LM).
@@ -32,13 +33,11 @@ from transformers import BartForConditionalGeneration, BartTokenizer
 tokenizer = BartTokenizer.from_pretrained('IDEA-CCNL/Yuyuan-Bart-139M')
 model = BartForConditionalGeneration.from_pretrained('IDEA-CCNL/Yuyuan-Bart-139M')
 
-origin_text = 'Results are presented of measuring fibrinogen fluorescence parameters in temperature range of 20-80 degrees C at different pH of the solution.'
-text = 'Results are presented of measuring fibrinogen fluorescence parameters in <mask> range of 20-80 degrees C at different pH of the <mask>.'
+text = 'Influenza is a <mask> disease.'
 input_ids = tokenizer([text], return_tensors="pt")['input_ids']
 model.eval()
 generated_ids = model.generate(
     input_ids=input_ids,
-    max_length=64,
 )
 preds = [tokenizer.decode(g, skip_special_tokens=True, clean_up_tokenization_spaces=True) for g in generated_ids]
 print(preds)