Update app.py

app.py

@@ -2,10 +2,10 @@ import streamlit as st
 from transformers import AutoModelForTokenClassification, pipeline
 
 # Chargement du modèle de NER
-model = AutoModelForTokenClassification.from_pretrained("roberta-base-ner")
+model = AutoModelForTokenClassification.from_pretrained("vonewman/wolof-finetuned-ner")
 
 # Définition de la pipeline
-pipeline = pipeline("ner", model=model)
+pipeline = pipeline("ner", model=model, aggregation_strategy="simple")
 
 # Fonction pour traiter un document
 def traiter_document(document):