Update app.py

app.py

@@ -28,12 +28,32 @@ binary_mapping = {
     'LABEL_1': 'hateful', 
 }
 
+# category_mapping = {
+#     'LABEL_0': 'non-hateful', 
+#     'LABEL_1': 'symbolization', 
+#     'LABEL_2': 'exaggeration/generalization/attribution/distortion', 
+#     'LABEL_3': 'swearing/insult/defamation/dehumanization',      
+#     'LABEL_4': 'threat of enmity/war/attack/murder/harm',     
+# }
+
 category_mapping = {
     'LABEL_0': 'non-hateful', 
-    'LABEL_1': 'symbolization', 
-    'LABEL_2': 'exaggeration/generalization/attribution/distortion', 
-    'LABEL_3': 'swearing/insult/defamation/dehumanization',      
-    'LABEL_4': 'threat of enmity/war/attack/murder/harm',     
+    'LABEL_1': 'symbolization/exaggeration/generalization/attribution/distortion', 
+    'LABEL_2': 'swearing/insult/defamation/dehumanization/threat of enmity/war/attack/murder/harm',   
+}
+
+target_mapping = {
+  'LABEL_0': 'No-group',
+  'LABEL_1': 'Refugees',
+  'LABEL_2': 'Israel-Jews',
+  'LABEL_3': 'Greeks',
+  'LABEL_4': 'Armenian',
+  'LABEL_5': 'Alevi',
+  'LABEL_6': 'Kurdish',
+  'LABEL_7': 'Arabian',
+  'LABEL_8': 'LGBTI+',
+  'LABEL_9': 'Women',
+  'LABEL_10': 'Other groups'
 }
 
 
@@ -52,13 +72,17 @@ def perform_binary_classification(input_text, selected_model):
     return binary_mapping.get(model(input_text)[0]['label'], 'error')
 
 def perform_categorization(input_text):
-    model = pipeline(model=f'gokceuludogan/berturk_tr_hateprint_cat_w0.1_b128')
+    model = pipeline(model='gokceuludogan/berturk_tr_hateprint_cat_class_w0.1_b128') # f'gokceuludogan/berturk_tr_hateprint_cat_w0.1_b128')
     return category_mapping.get(model(input_text)[0]['label'], 'error')
 
-def perform_target_detection(input_text):
-    model = pipeline(model='gokceuludogan/turna_generation_tr_hateprint_target')
-    return model(input_text)[0]['generated_text']
+# def perform_target_detection(input_text):
+#     model = pipeline(model='gokceuludogan/turna_generation_tr_hateprint_target')
+#     return model(input_text)[0]['generated_text']
 
+def perform_target_detection(input_text):
+    model = pipeline(model='gokceuludogan/berturk_tr_hateprint_target_class_w0.1') # f'gokceuludogan/berturk_tr_hateprint_cat_w0.1_b128')
+    return target_mapping.get(model(input_text)[0]['label'], 'error')
+    
 def perform_multi_detection(input_text):
     model = pipeline(model='gokceuludogan/turna_generation_tr_hateprint_multi')
     return model(input_text)[0]['generated_text']
@@ -77,7 +101,7 @@ with gr.Blocks(theme="abidlabs/Lime") as hate_speech_demo:
             model_choice_binary = gr.Radio(
                 choices=[
                     "turna_tr_hateprint_w0.1_new_", 
-                    "berturk_tr_hateprint_w0.1", 
+                    "berturk_tr_hateprint_w0.1_b128_v2", # "berturk_tr_hateprint_w0.1", 
                 ], 
                 label="Select Model", 
                 value="turna_tr_hateprint"