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,
|