Den4ikAI commited on
Commit
731e251
1 Parent(s): 0b9061a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -90,7 +90,7 @@ def punctuate(input_text, model_name):
90
  output = ""
91
  for item in preds:
92
  if item["word"] == ".":
93
- item["entity_group"] = "O"
94
  output += " " + process_token(item['word'].strip(), item['entity_group'])
95
 
96
  return output.strip()
 
90
  output = ""
91
  for item in preds:
92
  if item["word"] == ".":
93
+ item["entity_group"] = "LOWER_O"
94
  output += " " + process_token(item['word'].strip(), item['entity_group'])
95
 
96
  return output.strip()