Summarization / app.py
abdalrahmanshahrour's picture
Upload 3 files
5d711dd
raw
history blame
2.19 kB
from urllib.parse import unquote
import arabic_reshaper
import streamlit as st
from bidi.algorithm import get_display
from summarize import get_results
st.set_page_config(
page_title="Arabic Summarization",
page_icon="🤖",
layout="wide",
initial_sidebar_state="expanded",
menu_items={
'Get Help': 'https://www.extremelycoolapp.com/help',
'Report a bug': "https://www.extremelycoolapp.com/bug",
'About': "# Arabic Text Summarizeation , abdalrahman shahrour",
}
)
rtl = lambda w: get_display(f"{arabic_reshaper.reshape(w)}")
st.header('Arabic Text Summarization')
st.markdown("")
model = st.sidebar.selectbox('Select one', ['arabartsummarization', 'AraBART', 'auto-arabic-summarization', 'BERT2BERT', 'xlmroberta2xlmroberta', 'nltk_summarizer'],help="Model",)
# ahmeddbahaa/xlmroberta2xlmroberta-finetune-summarization-ar
st.sidebar.write("\n")
num_beams = st.sidebar.slider(
"Number of beams", min_value=1, max_value=10, value=3, step=1
)
st.sidebar.write("\n")
length_penalty = st.sidebar.slider(
"Length penalty ", min_value=0.1, max_value=3.0, value=1.0, step=0.1,
)
number_of_sentence = st.sidebar.slider(
"Number of sentence", min_value=1, max_value=10, value=3, step=1
)
doc = st.text_area("Enter the text to be summarized",height=550,value=" شهدت مدينة طرابلس، مساء أمس الأربعاء، احتجاجات شعبية وأعمال شغب لليوم الثالث على التوالي، وذلك بسبب تردي الوضع المعيشي والاقتصادي. واندلعت مواجهات عنيفة وعمليات كر وفر ما بين الجيش اللبناني والمحتجين استمرت لساعات، إثر محاولة فتح الطرقات المقطوعة، ما أدى إلى إصابة العشرات من الطرفين.")
summarize_button = st.button(label="🧞‍♂️ summarize 🧞‍♂️")
if summarize_button:
with st.spinner("جاري التلخيص ..."):
result = get_results(doc, model, num_beams, length_penalty,number_of_sentence)
if len(result) > 0:
st.write(result)
else:
st.write("")