Update README.md

README.md

@@ -70,6 +70,10 @@ pipeline_tag: translation
         inputs['attention_mask'] = inputs['attention_mask'][0][:-1].unsqueeze(dim=0)
 
     outputs = model.generate(**inputs, max_length=768, eos_token_id=tokenizer.eos_token_id)
+
+    input_len = len(inputs['input_ids'].squeeze())
+    translation = tokenizer.decode(outputs[0][input_len:], skip_special_tokens=True)
+    print(translation)
   ```
 
 ### Framework versions