Updating Demo Code
#2
by
sprice134
- opened
Hey there,
Your model is great, although I think that the demo code in the model card page is incorrect. On HuggingFace, to print the classification, the code is: print(output). However, this returns an encoded tensor. When you check the github however, the equivalent line is: print(tokenizer.decode(output[0])). Not a huge deal but might help others in the future so they don't have to go looking for it.
Thanks for pointing it out, I have updated the demo codes.
amphora
changed discussion status to
closed