JasonTPhillipsJr commited on
Commit
169e7aa
·
verified ·
1 Parent(s): fbaf643

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = tokenizer.decode(pseudo_sentence[0].tolist(), skip_special_tokens=False)
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,