predict() in Predict.py function issue "only one element tensors can be converted to Python scalars" about finbert HOT 4 CLOSED

I'm not sure what might be causing the problem. Could you please post the code that is causing the error and print logits right before you get the error? And also the complete error message.

from finbert.

I also get the same error. Please see the code I used and the logits before the error

from finbert.

When I modify the code like hlian1888 (logits = softmax(np.array(logits[0]))), I get these results which are different from the notebook on github

My results

from finbert.

@sharebear1993 sorry for the very late response. This problem should be fixed with the latest commit.

from finbert.