Update src/streamlit_app.py

src/streamlit_app.py

@@ -90,7 +90,7 @@ def load_ner_model():
         return pipeline(
             "token-classification",
             model="ml6team/keyphrase-extraction-kbir-inspec",
-            aggregation_strategy="max"
+            aggregation_strategy="max", stride=128, ignore_labels=["O"]
         )
     except Exception as e:
         st.error(f"Failed to load NER model: {e}")
@@ -138,11 +138,12 @@ if st.button("Results"):
                     # It just uses 'label'
                     data.append({
                         'word': entity['word'],
-                        'label': entity['label'],
+                        'label': entity['entity_group'], # This is the correct key
                         'score': entity['score'],
                         'start': entity['start'],
                         'end': entity['end']
                     })
+                    
 
                 if not data:
                     st.warning("No keyphrases found in the text.")