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/robeczech-base-binary-cs-iib', num_labels=2).to("cuda")
|
| 42 |
|
| 43 |
tokenizer = AutoTokenizer.from_pretrained(
|
| 44 |
+
'csocsci/robeczech-base-binary-cs-iib',
|
| 45 |
use_fast=False, truncation_side='left')
|
| 46 |
assert tokenizer.truncation_side == 'left'
|
| 47 |
|