Taja Kuzman
commited on
Commit
·
15883fb
1
Parent(s):
746aba2
Update README.md
Browse files
README.md
CHANGED
|
@@ -155,7 +155,7 @@ model_args= {
|
|
| 155 |
"silent": True
|
| 156 |
}
|
| 157 |
model = ClassificationModel(
|
| 158 |
-
"xlmroberta", "classla/xlm-roberta-base-multilingual-text-
|
| 159 |
args=model_args
|
| 160 |
|
| 161 |
)
|
|
|
|
| 155 |
"silent": True
|
| 156 |
}
|
| 157 |
model = ClassificationModel(
|
| 158 |
+
"xlmroberta", "classla/xlm-roberta-base-multilingual-text-genre-classifier", use_cuda=True,
|
| 159 |
args=model_args
|
| 160 |
|
| 161 |
)
|