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(",")]
|