better classification

app.py

@@ -48,7 +48,7 @@ elif input_method == "Write or Paste New Text":
 def setModel(model_checkpoint, aggregation):
     tokenizer = AutoTokenizer.from_pretrained("dsfsi/PuoBERTa-NER")
     model = AutoModelForTokenClassification.from_pretrained("dsfsi/PuoBERTa-NER")
-    return pipeline("token-classification", model="dsfsi/PuoBERTa-NER")
+    return pipeline("token-classification", model=model, tokenizer=tokenizer, aggregation_strategy=aggregation)
 
 @st.cache_resource
 def get_html(html: str):