updated model
Browse files
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/
|
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
|
36 |
|
37 |
description = """
|
38 |
-
- The model used for Recognizing entities [BERT-BASE-NER](https://huggingface.co/kazalbrur/
|
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 = '''
|