Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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(",")]
|