5roop commited on
Commit
9090130
1 Parent(s): 3c3e1a4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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="5roop/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
 
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