Mhammad Ibrahim commited on
Commit
f781087
·
1 Parent(s): 061f0ef
Files changed (2) hide show
  1. app.py +0 -1
  2. requirements.txt +0 -2
app.py CHANGED
@@ -14,7 +14,6 @@ from transformers import AutoTokenizer, AutoModelForTokenClassification, pipelin
14
  tokenizer = AutoTokenizer.from_pretrained("Mhammad2023/bert-finetuned-ner")
15
  model = AutoModelForTokenClassification.from_pretrained(
16
  "Mhammad2023/bert-finetuned-ner",
17
- from_tf=True,
18
  torch_dtype=torch.float32
19
  )
20
  classifier = pipeline("token-classification", model=model, tokenizer=tokenizer, device="cpu")
 
14
  tokenizer = AutoTokenizer.from_pretrained("Mhammad2023/bert-finetuned-ner")
15
  model = AutoModelForTokenClassification.from_pretrained(
16
  "Mhammad2023/bert-finetuned-ner",
 
17
  torch_dtype=torch.float32
18
  )
19
  classifier = pipeline("token-classification", model=model, tokenizer=tokenizer, device="cpu")
requirements.txt CHANGED
@@ -2,6 +2,4 @@ gradio>=4.0.0
2
  transformers>=4.21.0
3
  torch>=1.12.0
4
  tokenizers>=0.13.0
5
- tensorflow>=2.8.0
6
- tf-keras
7
  accelerate
 
2
  transformers>=4.21.0
3
  torch>=1.12.0
4
  tokenizers>=0.13.0
 
 
5
  accelerate