Update README.md

README.md

@@ -1,10 +1,13 @@
 ---
-license: apache-2.0
+license: mit
 tags:
 - setfit
 - sentence-transformers
 - text-classification
 pipeline_tag: text-classification
+language:
+- multilingual
+---
 ---
 
 # test_depres
@@ -29,8 +32,11 @@ from setfit import SetFitModel
 
 # Download from Hub and run inference
 model = SetFitModel.from_pretrained("test_depres")
+
+dict ={0:"positive", 1:"negative"}
 # Run inference
-preds = model(["i loved the spiderman movie!", "pineapple on pizza is the worst 🤮"])
+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])
 ```
 
 ## BibTeX entry and citation info
@@ -47,3 +53,13 @@ year = {2022},
 copyright = {Creative Commons Attribution 4.0 International}
 }
 ```
+## Citing & Authors
+```
+@misc{Uaritm,
+      title={SetFit: Classification of medical texts}, 
+      author={Vitaliy Ostashko},
+      year={2022},
+      url={https://esemi.org}
+}
+
+<!--- Describe where people can find more information -->