Update app.py

app.py

@@ -5,6 +5,11 @@ from transformers import AutoTokenizer, AutoModelForMaskedLM
 import pandas as pd
 import string
 from time import time
+from PIL import Image
+
+
+image = Image.open('./Image_AraThon.jpg')
+st.image(image)
 
 st.title("المساعدة اللغوية في التنبؤ بالمتلازمات والمتصاحبات وتصحيحها")
 default_value = "بيعت الأسلحة في السوق"
@@ -97,7 +102,7 @@ if (st.checkbox('الاستعانة بالرسم البياني', value=False)):
             break
     res_df = pd.DataFrame.from_dict(dict_res)
     res_df.index += 1
-    st.dataframe(res_df)
+    
     b = time()
     exec_time = (b-a)
     
@@ -105,7 +110,12 @@ if (st.checkbox('الاستعانة بالرسم البياني', value=False)):
     dict_next_words = next_word(text_st, pipe)
     df = pd.DataFrame.from_dict(dict_next_words)
     df.reset_index(drop=True, inplace=True)
+    
+    st.write("الكلمات المقترحة باستعمال النموذج اللغوي")
     st.dataframe(df)
+    st.markdown("""---""")
+    st.write("الكلمات المقترحة باستعمال الرسم البياني")
+    st.dataframe(res_df)
     
-    st.write(f'{str(exec_time)[:2] } :الوقت المستغرق باستعمال الرسم البياني')
-
+    
+    st.write(f'{str(exec_time)[:2] } :الوقت المستغرق باستعمال الرسم البياني')