Add accelerate package and simplify model loading

app.py

@@ -15,8 +15,7 @@ tokenizer = AutoTokenizer.from_pretrained("Mhammad2023/bert-finetuned-ner")
 model = AutoModelForTokenClassification.from_pretrained(
     "Mhammad2023/bert-finetuned-ner", 
     from_tf=True,
-    torch_dtype=torch.float32,
-    device_map="cpu"
+    torch_dtype=torch.float32
 )
 classifier = pipeline("token-classification", model=model, tokenizer=tokenizer, device="cpu")
 

requirements.txt

@@ -3,4 +3,5 @@ transformers>=4.21.0
 torch>=1.12.0
 tokenizers>=0.13.0
 tensorflow>=2.8.0
-tf-keras
+tf-keras
+accelerate