Mhammad Ibrahim commited on
Commit
1176a10
·
1 Parent(s): 589dc9a

Fix model loading from TensorFlow weights

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ import gradio as gr
10
  from transformers import pipeline
11
 
12
  # Load model from Hugging Face Hub
13
- classifier = pipeline("text-classification", model="Mhammad2023/bert-finetuned-ner")
14
 
15
  def predict(text):
16
  result = classifier(text)[0]
 
10
  from transformers import pipeline
11
 
12
  # Load model from Hugging Face Hub
13
+ classifier = pipeline("text-classification", model="Mhammad2023/bert-finetuned-ner", from_tf=True)
14
 
15
  def predict(text):
16
  result = classifier(text)[0]