diff --git a/demo/sequence_labeling/README.md b/demo/sequence_labeling/README.md index 5453b3a5046be6f7815c2292fe63dee490336cf7..bef869024486d30635c28f0d0a494332d0223f1b 100644 --- a/demo/sequence_labeling/README.md +++ b/demo/sequence_labeling/README.md @@ -183,7 +183,7 @@ model = hub.Module( label_map=label_map, ) -results = model.predict(data, max_seq_len=50, batch_size=1, use_gpu=False) +results = model.predict(data, max_seq_len=128, batch_size=1, use_gpu=True) for idx, text in enumerate(text_a): print(f'Data: {text} \t Lable: {", ".join(results[idx][1:len(text)+1])}') ```