Explicitly load model with from_tf=True

app.py

@@ -7,10 +7,12 @@
 # demo.launch()
 
 import gradio as gr
-from transformers import pipeline
+from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline
 
-# Load model from Hugging Face Hub
-classifier = pipeline("token-classification", model="Mhammad2023/bert-finetuned-ner", from_tf=True)
+# 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)
 
 def predict(text):
     results = classifier(text)

requirements.txt

@@ -1,4 +1,5 @@
 gradio>=4.0.0
 transformers>=4.21.0
 torch>=1.12.0
-tokenizers>=0.13.0
+tokenizers>=0.13.0
+tensorflow>=2.8.0