| import gradio as gr | |
| from transformers import AutoTokenizer, AutoModelForSequenceClassification | |
| import torch | |
| from collections import Counter | |
| from scipy.special import softmax | |
| article_string = "Author: <a href=\"https://huggingface.co/ruanchaves\">Ruan Chaves Rodrigues</a>. Read more about our <a href=\"https://github.com/ruanchaves/eplm\">research on the evaluation of Portuguese language models</a>." | |
| app_title = "Offensive Language Detection (Detecção de Linguagem Ofensiva)" | |
| app_description = """ | |
| This app detects offensive language on Portuguese text using multiple models. You can either introduce your own sentences by filling in "Text" or click on one of the examples provided below. | |
| (Este aplicativo detecta linguagem ofensiva em texto em português usando vários modelos. Introduza suas próprias frases preenchendo o campo "Text", ou clique em um dos exemplos fornecidos abaixo.) | |
| """ | |
| app_examples = [ | |
| ["Aquele cara é um babaca."], | |
| ["Quem não deve não teme!!"], | |
| ["Que nojo!🤮🤮🤮🤮🤮"], | |
| ["Vagabunda,Ordinária"], | |
| ["Vou mandar um óleo de peroba pra ela de presente! 😂😂😂😂"], | |
| ["Porque é corrupta é conivente com o desgoverno anterior"], | |
| ["A cada dia fico mais admirado com a cara de pau da elite dominante desse mundo até quando irão nos fazer de otários"] | |
| ] | |
| output_textbox_component_description = """ | |
| This box will display offensive language detection results based on the average score of multiple models. | |
| (Esta caixa exibirá resultados da detecção de linguagem ofensiva com base na pontuação média de vários modelos.) | |
| """ | |
| output_json_component_description = { "breakdown": """ | |
| This box presents a detailed breakdown of the evaluation for each model. | |
| """, | |
| "detalhamento": """ | |
| (Esta caixa apresenta um detalhamento da avaliação para cada modelo.) | |
| """ } | |
| short_score_descriptions = { | |
| 0: "Not offensive", | |
| 1: "Offensive" | |
| } | |
| score_descriptions = { | |
| 0: "This text is not offensive.", | |
| 1: "This text is offensive.", | |
| } | |
| score_descriptions_pt = { | |
| 1: "(Este texto é ofensivo.)", | |
| 0: "(Este texto não é ofensivo.)", | |
| } | |
| model_list = [ | |
| "ruanchaves/mdeberta-v3-base-hatebr", | |
| "ruanchaves/bert-base-portuguese-cased-hatebr", | |
| "ruanchaves/bert-large-portuguese-cased-hatebr", | |
| ] | |
| user_friendly_name = { | |
| "ruanchaves/mdeberta-v3-base-hatebr": "mDeBERTa-v3 (HateBR)", | |
| "ruanchaves/bert-base-portuguese-cased-hatebr": "BERTimbau base (HateBR)", | |
| "ruanchaves/bert-large-portuguese-cased-hatebr": "BERTimbau large (HateBR)", | |
| } | |
| reverse_user_friendly_name = { v:k for k,v in user_friendly_name.items() } | |
| user_friendly_name_list = list(user_friendly_name.values()) | |
| model_array = [] | |
| for model_name in model_list: | |
| row = {} | |
| row["name"] = model_name | |
| row["tokenizer"] = AutoTokenizer.from_pretrained(model_name) | |
| row["model"] = AutoModelForSequenceClassification.from_pretrained(model_name) | |
| model_array.append(row) | |
| def most_frequent(array): | |
| occurence_count = Counter(array) | |
| return occurence_count.most_common(1)[0][0] | |
| def predict(s1, chosen_model): | |
| if not chosen_model: | |
| chosen_model = user_friendly_name_list[0] | |
| scores = {} | |
| full_chosen_model_name = reverse_user_friendly_name[chosen_model] | |
| for row in model_array: | |
| name = row["name"] | |
| if name != full_chosen_model_name: | |
| continue | |
| else: | |
| tokenizer = row["tokenizer"] | |
| model = row["model"] | |
| model_input = tokenizer(*([s1],), padding=True, return_tensors="pt") | |
| with torch.no_grad(): | |
| output = model(**model_input) | |
| logits = output[0][0].detach().numpy() | |
| logits = softmax(logits).tolist() | |
| break | |
| def get_description(idx): | |
| description = score_descriptions[idx] | |
| description_pt = score_descriptions_pt[idx] | |
| final_description = description + "\n \n" + description_pt | |
| return final_description | |
| max_pos = logits.index(max(logits)) | |
| markdown_description = get_description(max_pos) | |
| scores = { short_score_descriptions[k]:v for k,v in enumerate(logits) } | |
| return scores, markdown_description | |
| inputs = [ | |
| gr.Textbox(label="Text", value=app_examples[0][0]), | |
| gr.Dropdown(label="Model", choices=user_friendly_name_list, value=user_friendly_name_list[0]) | |
| ] | |
| outputs = [ | |
| gr.Label(label="Result"), | |
| gr.Markdown(), | |
| ] | |
| gr.Interface(fn=predict, inputs=inputs, outputs=outputs, title=app_title, | |
| description=app_description, | |
| examples=app_examples, | |
| article = article_string).launch() |