Update app.py

app.py

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