SoDehghan's picture
Update models/hsd_tr.py
71a4d6b verified
raw
history blame
915 Bytes
import awesome_streamlit as ast
import streamlit as st
import requests
import time
from transformers import pipeline
import os
# 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}")