Spaces:
Sleeping
Sleeping
Update models/hsd_tr.py
Browse files- 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 |
|