Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -61,7 +61,7 @@ def process_entity(batch, model, device):
|
|
| 61 |
pseudo_sentence = batch['pseudo_sentence'].to(device)
|
| 62 |
|
| 63 |
# Convert tensor to list of token IDs, and decode them into a readable sentence
|
| 64 |
-
pseudo_sentence_decoded =
|
| 65 |
|
| 66 |
with torch.no_grad():
|
| 67 |
outputs = spaBERT_model(#input_ids=input_ids,
|
|
|
|
| 61 |
pseudo_sentence = batch['pseudo_sentence'].to(device)
|
| 62 |
|
| 63 |
# Convert tensor to list of token IDs, and decode them into a readable sentence
|
| 64 |
+
pseudo_sentence_decoded = bert_tokenizer.decode(pseudo_sentence[0].tolist(), skip_special_tokens=False)
|
| 65 |
|
| 66 |
with torch.no_grad():
|
| 67 |
outputs = spaBERT_model(#input_ids=input_ids,
|