Update README.md

README.md

@@ -1,13 +1,14 @@
 ---
 license: mit
-pipeline_tag: text-classification
 widget:
 - text: "whaling is part of the culture of various indigenous population and should be allowed for the purpose of maintaining this tradition and way of life and sustenance, among other uses of a whale. against We should ban whaling"
 ---
 
-
 ## Model Usage
 
+This model is built on custom code. So the inference api cannot be used directly. To use the model please follow the steps below... 
+
+
 ```python
 
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
@@ -29,14 +30,14 @@ encoding = tokenizer.encode_plus(
 
 with torch.no_grad():
         test_prediction = trained_model(encoding["input_ids"], encoding["attention_mask"])
-        test_prediction = test_prediction["logits"].flatten().numpy()
+        test_prediction = test_prediction["output"].flatten().numpy()
 
 ```
 
 ## Prediction
 To make a prediction and map the the outputs to the correct labels.
 During the competiton a threshold of 0.25 was used to binarize the output. 
-```
+```python
 THRESHOLD = 0.25
 LABEL_COLUMNS = ['Self-direction: thought','Self-direction: action','Stimulation','Hedonism','Achievement','Power: dominance','Power: resources','Face','Security: personal',
                  'Security: societal','Tradition','Conformity: rules','Conformity: interpersonal','Humility','Benevolence: caring','Benevolence: dependability','Universalism: concern','Universalism: nature','Universalism: tolerance','Universalism: objectivity']