emeraldgoose
/

bert-base-v1-sports

Inference Endpoints

Model card Files Files and versions Community

emeraldgoose commited on Nov 21, 2021

Commit

1d63d56

·

1 Parent(s): b21f26f

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -21,6 +21,6 @@ inputs = tokenizer.encode(text, return_tensors='pt')
 model.eval()
 outputs = model(inputs)['logits']
-predict = outputs.argmax(-1)
-print([tokenizer.decode(iter) for iter in predict])
 ```

 model.eval()
 outputs = model(inputs)['logits']
+predict = outputs.argmax(-1)[0]
+print(tokenizer.decode(predict))
 ```