vives commited on
Commit
06184f0
·
1 Parent(s): aab02c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -55,6 +55,8 @@ def concat_tokens_tags(sentences):
55
  tokens['input_ids'] = torch.stack(tokens['input_ids'])
56
  tokens['attention_mask'] = torch.stack(tokens['attention_mask'])
57
  return tokens
 
 
58
 
59
  if tags:
60
  tags = [x.lower().strip() for x in tags.split(",")]
 
55
  tokens['input_ids'] = torch.stack(tokens['input_ids'])
56
  tokens['attention_mask'] = torch.stack(tokens['attention_mask'])
57
  return tokens
58
+
59
+ # Process tags
60
 
61
  if tags:
62
  tags = [x.lower().strip() for x in tags.split(",")]