Spaces:
Running
Running
import streamlit as st | |
import requests | |
import time | |
from transformers import pipeline | |
import os | |
# Set the page configuration | |
st.set_page_config( | |
page_title="Hate Speech Detection", | |
page_icon="📖", #":bar_chart:" | |
layout='centered' | |
) | |
#title = r"$\textsf{\small Hate Speech Detection}$" | |
#st.title(title) | |
#st.write("In this HuggingFace space you will be able to use our Hate Speech Detection model built at [VERİM - Center of Excellence in Data Analytics - Sabanci University](https://github.com/verimsu).") | |
# Turkish | |
sentiment_pipeline_tr = pipeline(task = "text-classification", model = "SoDehghan/BERTurk-hate-speech") # "gritli/bert-sentiment-analyses-imdb" | |
header_tr = r"$\textsf{\scriptsize HSD in Turkish}$" | |
st.subheader(header_tr) | |
tr_input = st.text_area("Enter your text here:", height=50, key="tr_input") #height=30 | |
if st.button("Click for predictions!", key="tr_predict"): | |
st.write(" ") | |
with st.spinner('Generating predictions...'): | |
result_tr = sentiment_pipeline_tr(tr_input) | |
sentiment_tr = result_tr[0]["label"] | |
label_dict = {'LABEL_1': 'Hate ❌ ', 'LABEL_0': 'Non-hate ✅ '} #🚫 | |
sentiment_tr = label_dict[sentiment_tr] | |
strength_tr = " " | |
st.write(f"Detection: {sentiment_tr}, Strength: {strength_tr}") | |
st.write(" ") | |
# Arabic | |
sentiment_pipeline_ar = pipeline(task = "text-classification", model = "SoDehghan/BERTurk-hate-speech") | |
header_ar = r"$\textsf{\scriptsize HSD in Arabic}$" | |
st.subheader(header_ar) | |
ar_input = st.text_area("Enter your text here:", height=50 , key="ar_input") | |
if st.button("Click for predictions!", key="ar_predict"): | |
with st.spinner('Generating predictions...'): | |
result_ar = sentiment_pipeline_ar(ar_input) | |
sentiment_ar = result_ar[0]["label"] | |
label_dict = {'LABEL_1': 'Hate ❌ ', 'LABEL_0': 'Non-hate ✅ '} | |
sentiment_ar = label_dict[sentiment_ar] | |
strength_ar = " " | |
st.write(f"Detection: {sentiment_ar}, Strength: {strength_ar}") | |
st.sidebar.title("Hate Speech Detection") | |
#st.sidebar.write("In this HuggingFace space you can use Hate Speech Detection model built at [VERİM - Center of Excellence in Data Analytics - Sabanci University](https://github.com/verimsu).") | |
st.sidebar.write('This tool is developed in the context of the EU project "Utilizing Digital Technology for Social Cohesion, Positive Messaging and Peace by Boosting Collaboration, Exchange and Solidarity" (EuropeAid/170389/DD/ACT/Multi) by [Sabanci University Center of Excellence in Data Analytics](https://github.com/verimsu).') |