kazalbrur commited on
Commit
6cfb1b3
·
verified ·
1 Parent(s): 89188a9
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ def merge_tokens(tokens: List[Dict[str, any]]) -> List[Dict[str, any]]:
31
  return merged_tokens
32
 
33
  # Initialize Model
34
- get_completion = pipeline("ner", model="kazalbrur/bangla-english-med-bert-ner", device=0)
35
 
36
  @spaces.GPU(duration=120)
37
  def ner(input: str) -> Dict[str, Any]:
 
31
  return merged_tokens
32
 
33
  # Initialize Model
34
+ get_completion = pipeline("ner", model="kazalbrur/bangla-english-med-bert-ner")
35
 
36
  @spaces.GPU(duration=120)
37
  def ner(input: str) -> Dict[str, Any]: