Update README.md
Browse files
README.md
CHANGED
@@ -23,7 +23,7 @@ The model was fine-tuned on a Czech dataset of Instant Messenger dialogs of Adol
|
|
23 |
|
24 |
<!-- Provide the basic links for the model. -->
|
25 |
|
26 |
-
- **Repository:** https://github.com/
|
27 |
- **Paper:** Stay tuned!
|
28 |
|
29 |
## Usage
|
@@ -38,10 +38,10 @@ test_texts = ['Utterance1;Utterance2;Utterance3']
|
|
38 |
|
39 |
# Load the model and tokenizer
|
40 |
model = AutoModelForSequenceClassification.from_pretrained(
|
41 |
-
'
|
42 |
|
43 |
tokenizer = AutoTokenizer.from_pretrained(
|
44 |
-
'
|
45 |
use_fast=False, truncation_side='left')
|
46 |
assert tokenizer.truncation_side == 'left'
|
47 |
|
|
|
23 |
|
24 |
<!-- Provide the basic links for the model. -->
|
25 |
|
26 |
+
- **Repository:** https://github.com/csocsci/supportive-interactions
|
27 |
- **Paper:** Stay tuned!
|
28 |
|
29 |
## Usage
|
|
|
38 |
|
39 |
# Load the model and tokenizer
|
40 |
model = AutoModelForSequenceClassification.from_pretrained(
|
41 |
+
'csocsci/xlm-roberta-large-binary-cs-iib', num_labels=2).to("cuda")
|
42 |
|
43 |
tokenizer = AutoTokenizer.from_pretrained(
|
44 |
+
'csocsci/xlm-roberta-large-binary-cs-iib',
|
45 |
use_fast=False, truncation_side='left')
|
46 |
assert tokenizer.truncation_side == 'left'
|
47 |
|