kazalbrur commited on
Commit
28bded7
1 Parent(s): 1e08797

updated model

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -20,7 +20,7 @@ def merge_tokens(tokens: List[Dict[str, any]]) -> List[Dict[str, any]]:
20
  device = 0 if torch.cuda.is_available() else -1
21
 
22
  # Initialize Model
23
- get_completion = pipeline("ner", model="kazalbrur/bangla-english-med-bert-ner", device=device)
24
 
25
  @spaces.GPU(duration=120)
26
  def ner(input: str) -> Dict[str, Any]:
@@ -32,10 +32,10 @@ def ner(input: str) -> Dict[str, Any]:
32
  return {"text": input, "entities": [], "error": str(e)}
33
 
34
  ####### GRADIO APP #######
35
- title = """<h1 id="title"> Bangla Banglish and English Bio-Medical Entity Recognition </h1>"""
36
 
37
  description = """
38
- - The model used for Recognizing entities [BERT-BASE-NER](https://huggingface.co/kazalbrur/bangla-english-med-bert-ner).
39
  """
40
 
41
  css = '''
 
20
  device = 0 if torch.cuda.is_available() else -1
21
 
22
  # Initialize Model
23
+ get_completion = pipeline("ner", model="kazalbrur/BanglaMedNER", device=device)
24
 
25
  @spaces.GPU(duration=120)
26
  def ner(input: str) -> Dict[str, Any]:
 
32
  return {"text": input, "entities": [], "error": str(e)}
33
 
34
  ####### GRADIO APP #######
35
+ title = """<h1 id="title"> Bangla Bio-Medical Entity Recognition </h1>"""
36
 
37
  description = """
38
+ - The model used for Recognizing entities [BERT-BASE-NER](https://huggingface.co/kazalbrur/BanglaMedNER).
39
  """
40
 
41
  css = '''