Modify application file

app.py

@@ -28,12 +28,8 @@ elif selection == "convbert-base-turkish-cased-ner":
 elif selection == "xlm-roberta-base-turkish-ner":
     model_checkpoint = "akdeniz27/xlm-roberta-base-turkish-ner"
 
-if model_checkpoint != "akdeniz27/xlm-roberta-base-turkish-ner":
-    st.sidebar.header("Select Aggregation Strategy Type")
-    aggregation = st.sidebar.radio("", ('first', 'simple', 'average', 'max', 'none'))
-else:
-    aggregation = "none"
-    st.sidebar.write("Aggregation strategy is set as 'none' for xlm-roberta model")
+st.sidebar.header("Select Aggregation Strategy Type")
+aggregation = st.sidebar.radio("", ('first', 'simple', 'average', 'max', 'none'))
 
 st.header("Select Text Input Method")
 input_method = st.radio("", ('Select among Examples', 'Write or Paste New Text'))
@@ -58,7 +54,10 @@ if Run_Button == True:
     output = ner_pipeline(input_text)
 
     df = pd.DataFrame.from_dict(output)
-    cols_to_keep = ['word','entity_group','score','start','end']
+    if aggregation != "none":
+        cols_to_keep = ['word','entity_group','score','start','end']
+    else:
+        cols_to_keep = ['word','entity','score','start','end']
     df_final = df[cols_to_keep]
     
     st.header("Recognized Entities")