Added code to map outputs back to the labels

README.md

@@ -78,4 +78,20 @@ tokenizer = AutoTokenizer.from_pretrained("reachosen/autotrain-sdohv7-3701198597
 inputs = tokenizer("The Patient is homeless", return_tensors="pt")
 
 outputs = model(**inputs)
+
+# Extract logits from the outputs
+logits = outputs.logits
+
+# Apply softmax to get probabilities
+probabilities = F.softmax(logits, dim=1)
+
+# get class mapping
+configs_fp = "path to model files" + "config.json"
+with open(configs_fp, 'r') as configs_file:
+    class_mapping = json.load(configs_file)['id2label']
+
+# Create the class_probs dictionary
+class_probs = {class_mapping[str(i)]: probabilities[0][i].item() for i in range(len(probabilities[0]))}
+
+
 ```