uaritm
/

test_depres

Text Classification

sentence-transformers

Model card Files Files and versions Community

uaritm commited on Feb 16, 2023

Commit

f3891bd

·

1 Parent(s): 08b6570

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -35,8 +35,9 @@ model = SetFitModel.from_pretrained("test_depres")
 dict ={0:"positive", 1:"negative"}
 # Run inference
 preds = model(["What happened to me? I don't know what to do, where to go! Can anyone help me?"])
-dict.get(str(list(preds))[1:-1])
 ```
 ```

 dict ={0:"positive", 1:"negative"}
 # Run inference
 preds = model(["What happened to me? I don't know what to do, where to go! Can anyone help me?"])
+print(dict.get(preds.numpy()[0]))
 ```
 ```