TajaKuzman
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 |
)
|