Update README, add ds code

README.md

@@ -78,11 +78,9 @@ features = feature_extractor(
     padding=True,
     truncation=False)
 
-# convert to tensor
 inputs = torch.tensor(np.array(features['input_values']), dtype=torch.float32).to(device)
 attention_mask = torch.tensor(np.array(features['attention_mask']), dtype=torch.long).to(device)
 
-# apply model
 prob_dict = {}
 with torch.no_grad():
     model_output = model(inputs, attention_mask=attention_mask)