Hugging Face's logo

Spaces:
Dr-BERT
/
CAS-Biomedical-POS-Tagging
Runtime error

App Files Community
qanastek commited on
Commit
08720fd
·
1 Parent(s): 7fda840

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ if Run_Button == True:
43
 
44
  output_new = []
45
  for o in output:
46
- o["entity_group"] = id2label[str(o["entity_group"].split("_")[-1])]
47
  output_new.append(o)
48
 
49
  df = pd.DataFrame.from_dict(output_new)
 
43
 
44
  output_new = []
45
  for o in output:
46
+ o["entity_group"] = id2label[int(o["entity_group"].split("_")[-1])]
47
  output_new.append(o)
48
 
49
  df = pd.DataFrame.from_dict(output_new)