unikei commited on
Commit
d7fa639
1 Parent(s): f1a7871

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -1
README.md CHANGED
@@ -5,6 +5,9 @@ widget:
5
  datasets:
6
  - ChEMBL
7
  pipeline_tag: fill-mask
 
 
 
8
  ---
9
 
10
  # BERT base for SMILES
@@ -35,4 +38,4 @@ model = BertModel.from_pretrained(checkpoint)
35
  example = 'O=C([C@@H](c1ccc(cc1)O)N)N[C@@H]1C(=O)N2[C@@H]1SC([C@@H]2C(=O)O)(C)C'
36
  tokens = tokenizer(example, return_tensors='pt')
37
  predictions = model(**tokens)
38
- ```
 
5
  datasets:
6
  - ChEMBL
7
  pipeline_tag: fill-mask
8
+ tags:
9
+ - biology
10
+ - medical
11
  ---
12
 
13
  # BERT base for SMILES
 
38
  example = 'O=C([C@@H](c1ccc(cc1)O)N)N[C@@H]1C(=O)N2[C@@H]1SC([C@@H]2C(=O)O)(C)C'
39
  tokens = tokenizer(example, return_tensors='pt')
40
  predictions = model(**tokens)
41
+ ```