Fix meta tensor issue with explicit device mapping

app.py

@@ -7,12 +7,18 @@
 # demo.launch()
 
 import gradio as gr
+import torch
 from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline
 
 # Load model and tokenizer from Hugging Face Hub
 tokenizer = AutoTokenizer.from_pretrained("Mhammad2023/bert-finetuned-ner")
-model = AutoModelForTokenClassification.from_pretrained("Mhammad2023/bert-finetuned-ner", from_tf=True)
-classifier = pipeline("token-classification", model=model, tokenizer=tokenizer)
+model = AutoModelForTokenClassification.from_pretrained(
+    "Mhammad2023/bert-finetuned-ner", 
+    from_tf=True,
+    torch_dtype=torch.float32,
+    device_map="cpu"
+)
+classifier = pipeline("token-classification", model=model, tokenizer=tokenizer, device="cpu")
 
 def predict(text):
     results = classifier(text)