SoDehghan commited on
Commit
be29b3f
1 Parent(s): da6c664

Update models/hsd_ar.py

Browse files
Files changed (1) hide show
  1. models/hsd_ar.py +1 -1
models/hsd_ar.py CHANGED
@@ -14,7 +14,7 @@ import re
14
  #strength_pipeline_tr = pipeline(task = "text-classification", model = "SoDehghan/BERTurk-hate-speech-strength-prediction")
15
  #target_pipeline_tr = pipeline(task = "text-classification", model = "SoDehghan/test-target")
16
  strength_pipeline_tr = pipeline(task = "text-classification", model = "SoDehghan/hs-ar-translate-syn-4class-for-tool")
17
- target_pipeline_tr = pipeline(task = "text-classification", model = "SoDehghan/targ-ar-syn-for-tool", return_all_scores=True)
18
 
19
  AR_TWEETS_EXAMPLES = [
20
  "أريد أن يموت اللاجئون",
 
14
  #strength_pipeline_tr = pipeline(task = "text-classification", model = "SoDehghan/BERTurk-hate-speech-strength-prediction")
15
  #target_pipeline_tr = pipeline(task = "text-classification", model = "SoDehghan/test-target")
16
  strength_pipeline_tr = pipeline(task = "text-classification", model = "SoDehghan/hs-ar-translate-syn-4class-for-tool")
17
+ target_pipeline_tr = pipeline(task = "text-classification", model = "SoDehghan/targ-ar-translate-syn-for-tool", return_all_scores=True)
18
 
19
  AR_TWEETS_EXAMPLES = [
20
  "أريد أن يموت اللاجئون",