update

app.py

@@ -11,11 +11,11 @@ import torch
 from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline
 
 # Load model and tokenizer from Hugging Face Hub
-tokenizer = AutoTokenizer.from_pretrained("Mhammad2023/bert-finetuned-ner")
+tokenizer = AutoTokenizer.from_pretrained("Mhammad2023/bert-finetuned-ner-torch")
 model = AutoModelForTokenClassification.from_pretrained(
-    "Mhammad2023/bert-finetuned-ner"
+    "Mhammad2023/bert-finetuned-ner-torch"
 )
-classifier = pipeline("token-classification", model=model, tokenizer=tokenizer, device="cpu")
+classifier = pipeline("token-classification", model=model, tokenizer=tokenizer, aggregation_strategy="simple")
 
 def predict(text):
     results = classifier(text)