SoDehghan commited on
Commit
e7d08c1
·
verified ·
1 Parent(s): aecd298

Update models/hsd_tr.py

Browse files
Files changed (1) hide show
  1. models/hsd_tr.py +3 -1
models/hsd_tr.py CHANGED
@@ -6,7 +6,9 @@ import re
6
  #strength_pipeline_tr = pipeline(task = "text-classification", model = "SoDehghan/BERTurk-hate-speech-strength-prediction")
7
  #target_pipeline_tr = pipeline(task = "text-classification", model = "SoDehghan/test-target")
8
  strength_pipeline_tr = pipeline(task = "text-classification", model = "SoDehghan/hs-tr-syn-4class-for-tool")
9
- target_pipeline_tr = pipeline(task = "text-classification", model = "SoDehghan/targ-tr-syn-for-tool", return_all_scores=True)
 
 
10
  degree_pipeline_tr = pipeline(task = "text-classification", model = "SoDehghan/reg-hs-tr-syn-for-tool", function_to_apply="none")
11
  group_pipeline_tr = pipeline(task = "text-classification", model = "SoDehghan/group-hs-tr-for-tool")
12
 
 
6
  #strength_pipeline_tr = pipeline(task = "text-classification", model = "SoDehghan/BERTurk-hate-speech-strength-prediction")
7
  #target_pipeline_tr = pipeline(task = "text-classification", model = "SoDehghan/test-target")
8
  strength_pipeline_tr = pipeline(task = "text-classification", model = "SoDehghan/hs-tr-syn-4class-for-tool")
9
+ #target_pipeline_tr = pipeline(task = "text-classification", model = "SoDehghan/targ-tr-syn-for-tool", return_all_scores=True)
10
+ target_pipeline_tr = pipeline(task = "text-classification", model = "SoDehghan/test-push-to-hf", return_all_scores=True)
11
+
12
  degree_pipeline_tr = pipeline(task = "text-classification", model = "SoDehghan/reg-hs-tr-syn-for-tool", function_to_apply="none")
13
  group_pipeline_tr = pipeline(task = "text-classification", model = "SoDehghan/group-hs-tr-for-tool")
14