Update README.md
Browse files
README.md
CHANGED
@@ -25,7 +25,7 @@ The following snippet illustrates code usage:
|
|
25 |
!pip install mutual-implication-score
|
26 |
|
27 |
from mutual_implication_score import MIS
|
28 |
-
mis = MIS(device='cpu')
|
29 |
source_texts = ['I want to leave this room',
|
30 |
'Hello world, my name is Nick']
|
31 |
paraphrases = ['I want to go out of this room',
|
|
|
25 |
!pip install mutual-implication-score
|
26 |
|
27 |
from mutual_implication_score import MIS
|
28 |
+
mis = MIS(device='cpu')#cuda:0 for using cuda with certain index
|
29 |
source_texts = ['I want to leave this room',
|
30 |
'Hello world, my name is Nick']
|
31 |
paraphrases = ['I want to go out of this room',
|