StefanH commited on
Commit
963bbd5
·
1 Parent(s): d014508

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -32,7 +32,8 @@ You can use the model like this:
32
  >>> 'Add To Playlist', 'Book Restaurant', 'Get Weather', 'Play Music', 'Rate Book', 'Search Creative Work',
33
  >>> 'Search Screening Event'
34
  >>> ]
35
- >>> query = [[text, f'{lb} intent'] for lb in labels]
 
36
  >>> logits = model.predict(query, apply_softmax=True)
37
  >>> print(logits)
38
 
 
32
  >>> 'Add To Playlist', 'Book Restaurant', 'Get Weather', 'Play Music', 'Rate Book', 'Search Creative Work',
33
  >>> 'Search Screening Event'
34
  >>> ]
35
+ >>> aspect = 'intent'
36
+ >>> query = [[text, f'{lb} {aspect}'] for lb in labels]
37
  >>> logits = model.predict(query, apply_softmax=True)
38
  >>> print(logits)
39