Merge branch 'main' of https://huggingface.co/5roop/sloberta-frenk-hate into main

README.md

@@ -54,18 +54,17 @@ Comparison with `xlm-roberta-base`:
 ```python
 from simpletransformers.classification import ClassificationModel
 model_args = {
-        "num_train_epochs": 14,
-        "learning_rate": 1e-5,
-        "train_batch_size": 21,
-        }
+        "num_train_epochs": 6,
+        "learning_rate": 3e-6,
+        "train_batch_size": 69}
 
 model = ClassificationModel(
-    "bert", "5roop/bcms-bertic-frenk-hate", use_cuda=True,
+    "camembert", "5roop/sloberta-frenk-hate", use_cuda=True,
     args=model_args
     
 )
 
-predictions, logit_output = model.predict(["Silva, ti si grda in neprijazna.", "Naša hiša ima dimnik."])
+predictions, logit_output = model.predict(["Silva, ti si grda in neprijazna", "Naša hiša ima dimnik"])
 predictions
 ### Output:
 ### array([1, 0])