Update README.md

README.md

@@ -8,6 +8,9 @@ pipeline_tag: text-classification
 language:
 - fr
 library_name: setfit
+metrics:
+- f1
+- accuracy
 ---
 
 # binbin83/setfit-MiniLM-dialog-act-13nov
@@ -34,8 +37,11 @@ from setfit import SetFitModel
 
 # Download from Hub and run inference
 model = SetFitModel.from_pretrained("binbin83/setfit-MiniLM-dialog-act-13nov")
+label_dict = {'Introductory': 0, 'FollowUp': 1, 'Probing': 2, 'Specifying': 3, 'Structuring': 4, 'DirectQuestion': 5, 'Interpreting': 6, 'Ending': 7}
 # Run inference
 preds = model(["Vous pouvez continuer", "Pouvez-vous me dire précisément quel a été l'odre chronologique des événements ?"])
+labels = [[[f for f, p in zip(labels_dict, ps) if p] for ps in [pred]] for pred in preds ]
+
 ```
 
 ##  Labels and training data