Update README.md

README.md

@@ -27,12 +27,12 @@ print(model(**tokenized).logits)
 ```python
 from transformers import pipeline
 
-pipe = pipeline("text-classification", model="tum-nlp/NegBLEURT")
+pipe = pipeline("text-classification", model="tum-nlp/NegBLEURT", function_to_apply="none") # set function_to_apply="none" for regression output!
 pairwise_input = [
   [["Ray Charles is legendary.", "Ray Charles is a legend."]],
   [["Ray Charles is legendary.", "Ray Charles isn’t legendary."]]
 ]
-print(pipe(pairwise_input, function_to_apply="none"))
+print(pipe(pairwise_input))
 # returns [{'label': 'NegBLEURT_score', 'score': 0.8408917784690857}, {'label': 'NegBLEURT_score', 'score': 0.26007288694381714}]
 ```