Update README.md

README.md

@@ -2,17 +2,17 @@
 license: mit
 ---
 
-Example:
+Example classification:
 
 ```python
-  import torch
-  import tensorflow as tf
-  from transformers import AutoModelForSequenceClassification, AutoTokenizer
-  tokenizer = AutoTokenizer.from_pretrained("cardiffnlp/twitter-roberta-base-sentiment")
-  model = AutoModelForSequenceClassification.from_pretrained("paragon-analytics/bert_resil")
+import torch
+import tensorflow as tf
+from transformers import AutoModelForSequenceClassification, AutoTokenizer
+tokenizer = AutoTokenizer.from_pretrained("cardiffnlp/twitter-roberta-base-sentiment")
+model = AutoModelForSequenceClassification.from_pretrained("paragon-analytics/bert_resil")
 
-  encoded_input = tokenizer("We will survive this.", return_tensors='pt')
-  output = model(**encoded_input)
-  scores = output[0][0].detach().numpy()
-  scores = tf.nn.softmax(scores)
+encoded_input = tokenizer("We will survive this.", return_tensors='pt')
+output = model(**encoded_input)
+scores = output[0][0].detach().numpy()
+scores = tf.nn.softmax(scores)
 '''