Update README.md
Browse files
README.md
CHANGED
@@ -72,7 +72,7 @@ import torch
|
|
72 |
model_args = ClassificationArgs(
|
73 |
regression=True,
|
74 |
)
|
75 |
-
model = ClassificationModel(model_type="xlmroberta", model_name="
|
76 |
model.predict(["""Poštovani potpredsjedničke Vlade i ministre hrvatskih branitelja, mislite li
|
77 |
da ste zapravo iznevjerili svoje suborce s kojima ste 555 dana prosvjedovali
|
78 |
u šatoru protiv tadašnjih dužnosnika jer ste zapravo donijeli zakon koji je
|
|
|
72 |
model_args = ClassificationArgs(
|
73 |
regression=True,
|
74 |
)
|
75 |
+
model = ClassificationModel(model_type="xlmroberta", model_name="classla/xlm-r-parlasent",use_cuda=torch.cuda.is_available(), num_labels=1,args=model_args)
|
76 |
model.predict(["""Poštovani potpredsjedničke Vlade i ministre hrvatskih branitelja, mislite li
|
77 |
da ste zapravo iznevjerili svoje suborce s kojima ste 555 dana prosvjedovali
|
78 |
u šatoru protiv tadašnjih dužnosnika jer ste zapravo donijeli zakon koji je
|